JZOJ3019. 序列和(sum)
Description
N个数排成一个环,请选出不超过K段的连续的数,段与段间不能重叠,且使得选出的数和最大。
Input
第一行包含两个正整数N和k。
接下来1行描述这N个数。
Output
包含一个数,即要去的最大的和。
Sample Input
9 2
2 -1 2 -1 2 -4 1 -1 2
Sample Output
7
Data Constraint
20%:K=1,N<=1000
另外20%:K=1
另外20%:N<=1000
100%:K<=10,N<=100000
Solution
考试时想了一会儿就放弃了QWQ
其实就是一个DP
设f[i][j][k][l]表示到第i个数,已经用了j段,(k=0/1)是否选当前点,(l=0/1)开头点是否被选。
没有l的话简单DP
但由于是环,所以要加一些特殊处理:
先预处理个后缀和Sum[i],还有后缀和中的最大值Max[i]
这有什么用呢?
可以发现,只要是选了开头,那么从结尾开始向左延伸时肯定时越大越好,直到遇到当前点
所以在DP统计答案时,只要l为1,求ans时就加上Max[i+1]即可
Code
#include<bits/stdc++.h>
#define open_in(x) freopen(""#x".in","r",stdin);
#define open_out(x) freopen(""#x".out","w",stdout);
#define open_(x) freopen(""#x".in","w",stdout);
#define open(x) open_in(x);open_out(x);
#define mes(x,y) memset(x,y,sizeof(x));
#define mec(x,y) memcpy(x,y,sizeof(x));
using namespace std;
typedef long long ll;
typedef double db;
typedef unsigned long long ull;
const int X=100010;
const int S=11;
inline void read(ll &x)
{
int f=1;char ch=getchar();x=0;
while (ch<'0' || ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
x*=f;
}
ll n,m,a[X],f[X][S][2][2],Max[X],ans;
int main()
{
open(sum);
int i,j,k,l;
read(n),read(m);
for (i=1;i<=n;i++)
read(a[i]);
mes(Max,0);
for (ll i=n,sum=a[n];i>=1;sum+=a[--i])
Max[i]=max(Max[i+1],sum);
mes(f,128);
f[1][0][0][0]=0;
f[1][1][1][1]=a[1];
ans=max(0ll,a[1]);
for (i=2;i<=n;i++)
for (j=0;j<=m;j++)
{
ans=max(ans,(f[i][j][0][0]=max(f[i-1][j][0][0],f[i-1][j][1][0])));
ans=max(ans,(f[i][j][0][1]=max(f[i-1][j][0][1],f[i-1][j][1][1]))+Max[i+1]);
if (j>0)
{
ans=max(ans,(f[i][j][1][0]=max(f[i-1][j][1][0],f[i-1][j-1][0][0])+a[i]));
ans=max(ans,(f[i][j][1][1]=max(f[i-1][j][1][1],f[i-1][j-1][0][1])+a[i])+Max[i+1]);
}
}
printf("%lld\n",ans);
return 0;
}