思路来自SMCwwh:
1<= n <= 500000这么大的数据, 暴力是铁定tle的,
肯定每个人都知道要有线性算法才性.
我的思路是:
变量total表示所有符合条件的系列, 开始total=n;
从后往前扫描, 遇到负数就向前累加total--且flag[i]=1(表示已被访问过) , 直到和为正数为止.
这样遍历完整个数组,
如果到遍历完后累加和sum<=0, 则将sum再从数组最后开始累加直到和为正, 此时, 如果flag[i]=0则total--;
#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
__int64 a[500001];
bool flag[500001];
int main()
{
int t,n,i,total,id=1;
__int64 sum;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
memset(flag,0,sizeof(bool)*(n+1));
total=n;
for(i=0;i<n;i++)
scanf("%I64d",&a[i]);
sum=a[n-1];
if(sum<=0)
{
total--;
flag[n-1]=1;
}
for(i=n-2;i>=0;i--)
{
if(sum<=0)
{
sum+=a[i];
if(sum<=0)
{
total--;
flag[i]=1;
}
}
else
{
if(a[i]<=0)
{
sum=a[i];
total--;
flag[i]=1;
}
}
}
if(sum<=0)
{
for(i=n-1;i>=0;i--)
{
sum+=a[i];
if(sum<=0 && flag[i]==0)
total--;
if(sum>0)
break;
}
}
printf("Case %d: %d\n",id++,total);
}
return 0;
}