南阳oj742子串和在续(最大m段子序列和+动态规划)

来源:互联网 发布:山姆 知乎 编辑:程序博客网 时间:2024/06/06 16:09

子串和再续

时间限制: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段最大和
样例输入
12 6-1 4 -2 3 -2 4
样例输出
9

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




1 0
原创粉丝点击