题目链接:https://vjudge.net/contest/342715#problem/A
题意:给定由 n个整数(可能为负整数)组成的序列a1,a2,a3,……,an,以及一个正整数 m,要求确定序列 a1,a2,a3,……,an的 m个不相交子段,使这m个子段的总和达到最大,求出最大和。
输入:2 6 -1 4 -2 3 -2 3 输出:8
解释:n=6,m=2,对-1 4 -2 3 -2 3这6个数分2组,求2组和的最大值。
第一组:4 第二组:3 -2 3 答案:4+4=8
题目思路:动态规划+m子段和的最大值
初始:dp[i][j]
dp[i][j] 表示 第j个元素存放在第i组
决策:第j个元素是否放在第i组;
若放在第i组:dp[i][j]=dp[i][j-1]+s[i];
若不放在第i组:dp[i][j]=dp[i-1][k]+s[i];(k从第i个元素到第j个元素)
总结:dp[i][j]=max(dp[i][j-1],dp[i-1][k])+s[i]
讨论:假如我们采用这种写法去写的话,复杂度太高,三层:一层i,一层j,一层k
改进:一维数组dp[j]
讨论:dp[i-1][k]其实就是上一层i-1层元素到j-1的最大值,所以我们只需要标记一下即可
dp[j]=max(dp[j-1],pre[j-1]) +s[j] 其中:pre[j-1]的功能就是dp[i-1][k]
AC代码:
#include<bits/stdc++.h>
using namespace std;
#define inf 0x7f7f7f7f
#define maxn 1000010
int s[maxn],dp[maxn],pre[maxn];
int main(){
int n,m;//n个数字切成m段
while(scanf("%d%d",&m,&n)!=EOF){
for(int i=1;i<=n;i++){
cin>>s[i];
}
memset(dp,0,sizeof(dp));
memset(pre,0,sizeof(pre));
int max_dp;
for(int i=1;i<=m;i++){
max_dp=-inf;
for(int j=i;j<=n;j++){//注意:j的范围:从第i个到第n个
dp[j]=max(dp[j-1],pre[j-1])+s[j];
pre[j-1]=max_dp;
max_dp=max(max_dp,dp[j]);
}
}
cout<<max_dp<<endl;
}
}
WA代码:错误原因:对于第二个循环(关于dp的循环),当i=0,j=i-1=-1,此时会导致错误
#include<bits/stdc++.h>
#include<map>
using namespace std;
#define inf 0x7f7f7f7f
#define maxn 1000010
int s[maxn],dp[maxn],pre[maxn];
int main(){
int n,m;//n个数字切成m段
while(scanf("%d%d",&m,&n)!=EOF){
for(int i=0;i<n;i++){ //区别:i的取值
cin>>s[i];
}
memset(dp,0,sizeof(dp));
memset(pre,0,sizeof(pre));
int max_dp;
for(int i=0;i<m;i++){ //区别:i的取值
max_dp=-inf;
for(int j=i-1;j<n;j++){ //由于s[i]的下标因而改变j的取值
dp[j]=max(dp[j-1],pre[j-1])+s[j];
pre[j-1]=max_dp;
max_dp=max(max_dp,dp[j]);
}
}
cout<<max_dp<<endl;
}
}