首页 > 代码库 > CodeForces 747D Winter Is Coming
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;}
CodeForces 747D Winter Is Coming
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。