Description
给出一个长度为n的序列a,要求将a分成若干段,使得每一段作为一个子序列其所有前缀和非负
Input
多组用例,每组用例首先输入一个整数n表示序列长度,之后n个整数ai表示该序列,以文件为结束输入(1<=n<=10^6,-10000<=ai<=10000,输入保证有解)
Output
对于每组用例,输出所有合法解中的最大分割段数
Sample Input
6
1 2 3 4 5 6
4
1 2 -3 0
5
0 0 0 0 0
Sample Output
6
2
5
Solution
贪心,因为后面数被分成几段并不会影响前面的数,故从后往前扫,给出两个结论
1.每次从后往前扫遇到的第一个负数一定是某个段的末尾
如果这个负数是某一段中间值,由于其后面都是非负的,而且后面的数影响不到前面的前缀和,那么将后面的非负值单独拿出来作为一个段显然更优
2.每次从后往前扫遇见第一个负数之后,不妨令这个负数下标为i,往前找离i最近的j使得a[j]+…+a[i]>=0,从j到i作为一段显然是合法的,且是最优的,因为如果选择一个小于j的k,使得k到i作为合法的一段,显然不如分割成k到j-1作为一段,j到i作为一段(由k到i合法和j到i合法轻易知道k到j-1和j到i也是合法的)
Code
#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
#define maxn 1111111
int n,a[maxn];
ll sum[maxn];
int main()
{
while(~scanf("%d",&n))
{
sum[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+a[i];
}
int ans=0;
for(int i=n;i>=1;i--)
{
if(a[i]>=0)ans++;
else
{
ll temp=sum[i];
int j=i-1;
for(;j>=1;j--)
{
if(sum[j]>=temp)continue;
else if(sum[j]<temp)
{
i=j+1,ans++;
break;
}
}
if(j==0)
{
ans++;
break;
}
}
}
printf("%d\n",ans);
}
return 0;
}