Description
N个整数组成的序列a[1],a[2],a[3],…,a[n],将这N个数划分为互不相交的M个子段,并且这M个子段的和是最大的。如果M >= N个数中正数的个数,那么输出所有正数的和。
例如:-2 11 -4 13 -5 6 -2,分为2段,11 -4 13一段,6一段,和为26。
Solution
刷刷水有益身心健康。
不过我还是没有一眼看出来。
考虑DP的以为必须选和可选可不选这个思路还是很重要的。
很经典的水题。
Code
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=5007;
ll i,j,k,l,t,n,m,ans;
ll a[maxn],f[2][maxn],g[2][maxn],u,v;
int main(){
scanf("%lld%lld",&n,&m);
fo(i,1,n)scanf("%lld",&a[i]);
fo(i,0,1)fo(j,0,n)f[i][j]=g[i][j]=-9999999999999;
fo(i,1,n)f[0][i]=g[0][i]=0;
g[0][0]=0;
fo(i,1,m){
v=u^1;
fo(j,1,n){
f[v][j]=max(g[u][j-1]+a[j],f[v][j-1]+a[j]);
g[v][j]=max(f[v][j],g[v][j-1]);
}
u=v;
}
printf("%lld\n",g[u][n]);
}