HNU 12848 Energy Tycoon (贪心)

贪心

#include<bits/stdc++.h>using namespace std;int main(){    long long n;    char s[100020];    while(scanf("%I64d",&n)==1)    {        scanf("%s",s);        int len=strlen(s);        long long sum1=0,sum2=0,ans=0;        for(int j=0;j<len;j++)        {            if(s[j]=='1')            {                if(sum1+sum2*2+1<=n)                    sum1++;                else                {                    if(sum2>0)                    {                        sum2--;                        sum1++;                    }                }            }            else            {                if(sum1+sum2*2+2<=n)                    sum2++;            }            ans+=(sum1+sum2);        }        printf("%I64d\n",ans);    }    return 0;}