寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #stairs - 階段 (Stairs)

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;
}