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
;
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;
}