题面
显然每次操作贪心对最大的c[i]减一是正确的。但如果从大到小排序,减前面c[i]个并不能保证有序。例如4,3,2,2,2,1,c[i]=3,减得4,3,1,2,2,1。我们应该把第c[i]个数求出来,>c[i]的照常减,=c[i]的优先减靠右的即可。线段树维护。
听说还可以二分?
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=1000010;
int n,m,a[1000010];
int read()
{
int x=0;
char ch=getchar();
while(ch<'0'||ch>'9') ch=getchar();
while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();}
return x;
}
struct tree
{
int mi,mx,l,r,p;
tree *ls,*rs;
tree()
{
mi=mx=l=r=p=0;
ls=rs=NULL;
}
void cal(int add){mi+=add;mx+=add;}
void update()
{
mi=min(ls->mi,rs->mi);
mx=max(ls->mx,rs->mx);
}
void pushdown()
{
if(p!=0)
{
ls->p+=p;rs->p+=p;
ls->cal(p);rs->cal(p);
p=0;
}
}
void build(int lx,int rx)
{
l=lx;r=rx;
if(l==r) {mi=mx=a[l];return;}
int mid=(l+r)>>1;
(ls=new tree)->build(lx,mid);
(rs=new tree)->build(mid+1,rx);
update();
}
void dec(int lx,int rx)
{
if(l==lx&&r==rx) {p--;cal(-1);return;}
pushdown();
int mid=(l+r)>>1;
if(rx<=mid) ls->dec(lx,rx);
else if(lx>mid) rs->dec(lx,rx);
else ls->dec(lx,mid),rs->dec(mid+1,rx);
update();
}
int query(int x)
{
if(l==r) {return mi;}
pushdown();
int mid=(l+r)>>1;
if(x<=mid) return ls->query(x);
else return rs->query(x);
}
int lbd(int c)
{
if(mx<c) return l-1;
if(mi>c||l==r) return r;
pushdown();
if(rs->mx>=c) return rs->lbd(c);
else return ls->lbd(c);
}
}*xtr;
int main()
{
n=read();m=read();
for(int i=1;i<=n;i++)
a[i]=read();
sort(a+1,a+n+1,greater<int>());
(xtr=new tree)->build(1,n);
int rnd=1;
for(rnd=1;rnd<=m;rnd++)
{
int t=read(),o,r1,r2;
if(t>n) {rnd--;break;}
o=xtr->query(t);
r1=xtr->lbd(o+1);
r2=xtr->lbd(o);
if(r1>=1) xtr->dec(1,r1);
if(t-r1>=1) xtr->dec(r2-t+r1+1,r2);
if(xtr->mi<0) {rnd--;break;}
}
printf("%d",min(m,rnd));
return 0;
}