HDU 5783 Divide the Sequence(贪心)

来源:互联网 发布:数据统治世界 百度云 编辑:程序博客网 时间:2024/05/19 02:31

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 1111111int 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;}
0 0
原创粉丝点击