BZOJ P4378 「POI2015」Logistyka【树状数组】【离散化】

维护一个长度为n的序列，一开始都是0，支持以下两种操作：
1.U k a 将序列中第k个数修改为a。
2.Z c s 在这个序列上，每次选出c个正数，并将它们都减去1，询问能否进行s次操作。每次询问独立，即每次询问不会对序列进行修改。

考虑一下Z操作。

方便叙述，我们修改一下变量名：$x$表示每次选取的个数，$y$表示一共选取次数。

假设我们已经知道了当前数中$\ge y$的数有$z$个，如果当前$z\ge x$，那么显然有解，我们接下来考虑$z<x$的情况。

除去$z$，此时我们还需要进行$x-z$次选取，而所对应的值就是$(x-z)\ast y$，设$sum$为除去那$z$个数的总和值。

如果$sum\ge (x-z)\ast y$，那么我们一定可以继续执行到结束。

理解一下正确性：

如果我们得到了$sum\ge (x-z)\ast y$但是如果某次操作后选不出x-c个人，那么这$x-z$当中肯定存在一个数$>y$，就矛盾了。

另外，离散化一下。

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define ll long long
#define rep(i,x,y) for(ll i=(x);i<=(y);i++)
#define repd(i,x,y) for(ll i=(x);i>=(y);i--)
using namespace std;

const ll N=2e5+5;

ll n,m,a[N],b[N],c[N],d[N],type[N];

#define lowbit(x) (x&(-x))

struct node{
	ll bit[N];
	void update(ll x,ll y) {
		for(;x<=m;x+=lowbit(x)) bit[x]+=y;
	}
	
	ll getsum(ll x) {
		ll ret=0;
		for(;x;x-=lowbit(x)) ret+=bit[x];
		return ret;
	}
}t1,t2;

inline ll read() {
	ll x=0;char ch=getchar();bool f=0;
	while(ch>'9'||ch<'0'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return f?-x:x;
}

int main() {
	n=read(),m=read();
	
	rep(i,1,m) {
		char ch[2];
		scanf("%s",ch+1),a[i]=read(),b[i]=c[i]=read();
		
		if(ch[1]=='U') type[i]=1;
		else type[i]=2;
	}
	
	sort(c+1,c+1+m);
	
	rep(i,1,m) b[i]=lower_bound(c+1,c+1+m,b[i])-c;
	
	rep(i,1,m) {
		if(type[i]==1) {
			if(d[a[i]]) t1.update(d[a[i]],-1);t1.update(b[i],1);
			if(d[a[i]]) t2.update(d[a[i]],-c[d[a[i]]]);t2.update(b[i],c[b[i]]);d[a[i]]=b[i];
		} else {
			ll ans=t1.getsum(m)-t1.getsum(b[i]-1);
			
			if(ans>=a[i]) puts("TAK");
			else {
				if(t2.getsum(b[i]-1)>=(a[i]-ans)*c[b[i]]) puts("TAK");
				else puts("NIE");
			}
		}
	}
	
	return 0;
}