2010年 日本情報オリンピック春合宿OJから、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
「競技プログラミング」を略して、「競プロ」などと呼ばれています。
#stairs - 階段 (Stairs)
https://www.ioi-jp.org/camp/2010/2010-sp-tasks/2010-sp-day1_20.pdf
僕が作成、提出したコードは、以下のとおりです。
/* AtCoder Problems in C++ #stairs - 階段 (Stairs) https://www.ioi-jp.org/camp/2010/2010-sp-tasks/2010-sp-day1_20.pdf 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; const ll mod=1234567; int main(void){ ll n,p; cin>>n>>p; vector<ll> h(n); vector<ll> s(n+1); for(ll i=0;i<n;i++){ cin>>h[i]; s[i+1]=s[i]+h[i]; } vector<ll> dp(n+1),dps(n+2); dp[0]=1; dps[1]=dp[0]; for(ll i=1;i<n+1;i++){ ll idx=lower_bound(s.begin(),s.end(),s[i]-p)-s.begin(); dp[i]=(dps[i]-dps[idx])%mod; dps[i+1]=(dps[i]+dp[i])%mod; } if(dp[n]<0){ dp[n]+=mod; } cout<<dp[n]<<endl; return 0; }