アルゴ式(beta版)の「動的計画法 (DP):【補充】練習問題集」からの出典です。
algo-method.com
アルゴ式とは...
>・プログラミングや情報科学をコツコツ学べる「教科書」
>・学んだ内容をゲーム感覚で大量に実践できる「練習問題」
>の2つで構成される、Web上で完結した学習コンテンツです。
Q. 各区間の総和を大きくしたい
/* アルゴ式(beta版): C++による「動的計画法 (DP):【補充】練習問題集」- Q. 各区間の総和を大きくしたい Q. 各区間の総和を大きくしたい https://algo-method.com/tasks/1120 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; bool tf(ll x,ll n,ll k,vector<ll> &a){ vector<ll> v(n+1,0); for(ll i=0;i<n;i++){ v[i+1]=v[i]+a[i]; } vector<ll> dp(n+1,0); //dp[0]=0; for(ll i=0;i<n;i++){ for(ll j=0;j<=i;j++){ if(j!=0 && dp[j]==0){ continue; } if(v[i+1]-v[j]>=x){ dp[i+1]=max(dp[i+1],dp[j]+1); } } } return (dp[n]>=k); } int main(void){ ll n,k; cin>>n>>k; vector<ll> a(n); for(ll i=0;i<n;i++){ cin>>a[i]; } ll inf=1<<30,l=0,r=inf; while(abs(l-r)>1){ ll mid=(l+r)/2; if(tf(mid,n,k,a)){ l=mid; }else{ r=mid; } } cout<<l<<endl; return 0; }
設問の出典は、情報科学をコツコツ積み立てて学習できるサービス「アルゴ式(beta版)」です。
algo-method.com