woc第一遍题目看错T_T。以为是选出连续d个那还怎么做啊。。。
显然,如果存在>=s的数有k个,那么我们一定每次都要选这k个,如果k>=c就不用管了肯定TAK。
否则,考虑<s的数的数字和,如果数字和>=(c-k)*s,那么剩下的数就可以将剩余的空填完;否则不能。
考虑证明。数字和<(c-k)*s肯定不能,因为根本不够取;否则数的个数显然>=(c-k)*s/(s-1)>c-k,也就是每一次都一定能找出c-k个数。每次都从大的取,那么取完以后还满足。
于是可以用树状数组在离散化以后维护<=t的数的和,<=t的数的个数。
AC代码如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long
#define N 1000005
using namespace std;
int n,m,cnt,a[N],x[N],y[N],z[N],num[N],hash[N];
struct bit_node{
ll c[N];
void add(int x,int t){
for (; x<=cnt; x+=x&-x) c[x]+=t;
}
ll qry(int x){
ll t=0; for (; x; x-=x&-x) t+=c[x]; return t;
}
}bit1,bit2;
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;
}
int find(int x){
int l=1,r=cnt,mid;
while (l<r){
mid=(l+r)>>1;
if (hash[mid]<x) l=mid+1; else r=mid;
}
return l;
}
int main(){
n=read(); m=read(); int i,j; char ch;
for (i=1; i<=m; i++){
ch=getchar(); while (ch<'A' || ch>'Z') ch=getchar();
x[i]=read(); y[i]=num[i]=read();
z[i]=(ch=='U')?1:0;
}
sort(num+1,num+m+1); hash[cnt=1]=num[1];
for (i=2; i<=m; i++)
if (num[i]!=num[i-1]) hash[++cnt]=num[i];
for (i=1; i<=m; i++) y[i]=find(y[i]);
for (i=1; i<=m; i++) if (z[i]){
if (j=a[x[i]]){
bit1.add(j,-1); bit2.add(j,-hash[j]);
}
a[x[i]]=y[i];
bit1.add(y[i],1); bit2.add(y[i],hash[y[i]]);
} else puts(bit2.qry(y[i]-1)>=(x[i]-bit1.qry(cnt)+bit1.qry(y[i]-1))*hash[y[i]]?"TAK":"NIE");
return 0;
}
by lych
2016.3.3