子串和再续
时间限制:
1000 ms | 内存限制:
65535 KB
难度:
4
-
描述
-
给你一个序列 S1, S2, S3, S4 ... Sx, ... Sn (1 ≤ x ≤ n ≤ 1,000,000, -32768 ≤ Sx ≤ 32767). 我们定义
sum(i, j) = Si + ... + Sj (1 ≤ i ≤ j ≤ n).现在给你一个 m(8>m>0&&m<n)你的任务是计算
sum(i1, j1) + sum(i2, j2) + sum(i3, j3) + ... + sum(im, jm) ;我们规定他是不相交的。
请输出m段最大和,比如:m = 2,n = 6 ,{-1 4 -2 3 -2 4} 它的结果是 9;-
输入
-
输入 T,表示T组数据
第二行 分别是m,n;
输出
- 请输出m段最大和 样例输入
-
1 2 6 -1 4 -2 3 -2 4
样例输出
-
9
-
输入 T,表示T组数据
dp[i][j]表示用前j个数组成i段的最大和,其中第i段包括j。
则有:dp[i][j]=max(dp[i][j-1],dp[i-1][k])+a[j] ,其中i-1<k<j-1,注意k不能等于j-1.
用这个状态转移方程时间复杂度为O(m*n*n)明显会超时,仔细观察
dp[i][j+1]=max(dp[i][j]+dp[i-1][k])+a[j+1],i-1 <k<j,我们会发现 dp[i-1][k],当i-1<k<j-1时,我们
在求dp[i][j]的时候已经算过了,只需要在求出k=j-1的情况就可以,可得,dp[i][j+1]=max(dp[i][j],dp[i-1][j-1])+a[j+1],即得到状态转移方程:dp[i][j]=max(dp[i][j-1],dp[i-1][j-2])+a[j]。
此时的时间复杂度为 O(m*n),由于m的范围比较小,所以这里我没有采用滚动数组。
#include<stdio.h>
#include<string.h>
#define INF 1<<31-1
#define max(x,y)(x>y?x:y)
int dp[9][1000010];
int a[1000010];
int main()
{
int t,m,n,i,j,res;
scanf("%d",&t);
while(t--)
{
res=-INF;
memset(dp,0,sizeof(dp));
scanf("%d%d",&m,&n);
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
for(i=1;i<=m;i++)
dp[i][i]=dp[i-1][i-1]+a[i];
for(i=1;i<=m;i++)
{
for(j=i+1;j<=n;j++)
{
dp[i][j]=max(dp[i][j-1],dp[i-1][j-2])+a[j];
}
}
for(i=m;i<=n;i++)
res=max(res,dp[m][i]);
printf("%d\n",res);
}
return 0;
}