CodeForces 747D Winter Is Coming

贪心。

只考虑负数的位置,先填间隔较小的,再填间隔较大的。如果填不满就不填,如果有多余就留给最后一个负数到终点这段路。

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<vector>
#include<queue>
#include<algorithm>
#include<set>
using namespace std;

int n,k,a[200010],sz;
struct X
{
    int L,R,len;
} s[200010];
vector<int>v;
int f[200010];

bool cmp(X a, X b)
{
    return a.len<b.len;
}

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]);
    int sum=0; for(int i=1; i<=n; i++) if(a[i]<0) f[i]=1,sum++;
    if(sum>k) printf("-1\n");
    else
    {
        if(sum==0) printf("0\n");
        else
        {
            for(int i=1; i<=n; i++) if(a[i]<0) v.push_back(i);

            for(int i=0; i<v.size()-1; i++) s[sz].L=v[i], s[sz].R=v[i+1], s[sz].len = s[sz].R-s[sz].L-1, sz++;

            sort(s,s+sz,cmp);
            sum=k-sum;
            for(int i=0; i<sz; i++)
            {
                if(sum>=s[i].len)
                {
                    for(int j=s[i].L;j<=s[i].R;j++) f[j]=1;
                    sum=sum-s[i].len;
                }
            }

            if(sum>=n+1-v[v.size()-1]-1)
            {
                for(int i=v[v.size()-1];i<=n;i++) f[i]=1;
            }

          //  for(int i=1;i<=n;i++) printf("%d ",f[i]);

            int ans=0;

            int f1=1,f2=n;
            for(int i=1;i<=n;i++)
            {
                if(f[i]==0) continue;
                else {f1=i;break;}
            }

            for(int i=n;i>=1;i--)
            {
                if(f[i]==0) continue;
                else {f2=i;break;}
            }

            ans=ans+1;

            if(f2!=n) ans=ans+1;

            for(int i=f1+1;i<=f2;i++)
            {
                if(f[i]!=f[i-1]) ans++;
            }

            printf("%d\n",ans);
        }

    }
    return 0;
}

 

posted @ 2016-12-30 09:22 Fighting_Heart 阅读(...) 评论(...) 编辑 收藏