題意,長度爲n的序列,讓你求出k個長度爲m的序列的和最大。
思路:很明顯是DP,我們先求一下前綴和,然後這裏我們定義dp【i】【j】截止到i時,j個序列的和的最優解。然後我們就能得到狀態轉移方程
dp[ i ][ j ]=max(dp[ i - 1 ][ j ] , dp[ i - m ] [ j - 1]+sum[ i ]-sum[ i - m]);
最後的dp[ n ] [ k ]就是全局最優解。
代碼如下:
#include<bits/stdc++.h>
#include<bitset>
#include<unordered_map>
#define pb push_back
#define bp __builtin_popcount
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=5e3+100;
const int MOD=1e4+7;
const double PI=3.1415926535;
int lowbit(int x){return x&-x;}
inline ll dpow(ll a, ll b){ ll r = 1, t = a; while (b){ if (b & 1)r = (r*t) % MOD; b >>= 1; t = (t*t) % MOD; }return r; }
inline ll fpow(ll a, ll b){ ll r = 1, t = a; while (b){ if (b & 1)r = (r*t); b >>= 1; t = (t*t); }return r; }
ll dp[maxn][maxn],n,m,k,a[maxn],sum[maxn];
int main()
{
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum[i]=sum[i-1]+a[i];
}
for(int i=1;i<=n;i++)
for(int j=1;j<=k;j++)
{
dp[i][j]=dp[i-1][j];
if(i>=m)
dp[i][j]=max(dp[i-m][j-1]+sum[i]-sum[i-m],dp[i-1][j]);
}
cout<<dp[n][k]<<endl;
//system("pause");
return 0;
}