Codeforces 747D Winter is Coming 贪心(从最坏解开始优化)

点击打开链接

如果k>=cnt 一定有解 即仅在neg使用tire 答案最坏为cnt*2;
可以在positive上使用tire的次数最多为k-cnt天 
如果两个neg之间长度(positive个数)为l,则如果在l上依旧使用tire 则k-=l ans-=2
贪心:优先对l小用tire即可 
最后特判最后一个neg能否使用到结束即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+20; 
int t[N];
vector<int> l;
int main()
{
	int n,k;
	while(cin>>n>>k)
	{
		l.clear();
		int cnt=0,neg;//negative个数,位置 
		int pre=-1;
		for(int i=1;i<=n;i++)
		{
			scanf("%I64d",&t[i]);
			if(t[i]<0)
			{
				cnt++; 
				neg=i;//neg的最大位置
				if(pre==-1)
				pre=i;
				else
				{
					l.push_back(i-pre-1);//中间postive长度 
					pre=i;
				} 
			}
		}
		if(cnt==0)
		{
			puts("0");
			continue;
		}
		if(cnt>k)
		puts("-1"); 
		else
		{
			int ans=cnt*2;
			k-=cnt;
			
			sort(l.begin(),l.end()); 
			for(int i=0;i<l.size();i++)
			{
				if(k>=l[i])
				{
					k-=l[i];
					ans-=2;
				}
			}
			//特判 
			if(k>=n-neg)
			ans-=1;
				//k-=n-neg-1;		
			cout<<ans<<endl;
		}
		
	}
	return 0;
}