Bzoj4378 [poi2015]logistyka

离散化后用树状数组维护

Posted by yjjr's blog on April 11, 2018

标签:树状数组,离散化

题目

题目传送门

Description

维护一个长度为n的序列,一开始都是0,支持以下两种操作:

  1. U k a 将序列中第k个数修改为a。
  2. Z c s 在这个序列上,每次选出c个正数,并将它们都减去1,询问能否进行s次操作。

每次询问独立,即每次询问不会对序列进行修改。

Input

第一行包含两个正整数,分别表示序列长度和操作次数。 接下来m行为m个操作,其中

Output

包含若干行,对于每个Z询问,若可行,输出TAK,否则输出NIE。

Sample Input

3 8
U 1 5
U 2 7
Z 2 6
U 3 1
Z 2 6
U 2 2
Z 2 6
Z 2 1

Sample Output

NIE
TAK
NIE
TAK

分析

如果存在的数有k个,那么我们一定每次都要选这k个,如果k>=c就直接输出TAK

否则,考虑的数的数字和,如果数字和,那么剩下的数就可以将剩余的空填完

先离散化,然后用树状数组维护的数字个数和数的和

code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=last[x];i;i=e[i].next)
using namespace std;
inline ll read(){
	ll f=1,x=0;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
//**********head by yjjr**********
#define mid ((l+r)>>1)
#define lowbit(x) (x&(-x))
const int maxn=1e6+6;
int n,m,cnt,j,a[maxn],x[maxn],y[maxn],z[maxn],num[maxn],hash[maxn];
char ch[10];
struct bit_node{
	ll c[maxn];
	void modify(int x,int t){for(;x<=cnt;x+=lowbit(x))c[x]+=t;}
	ll query(int x){ll re=0;for(;x;x-=lowbit(x))re+=c[x];return re;}
}bit1,bit2;
inline int find(int x){
	int l=1,r=cnt;
	while(l<r)if(hash[mid]<x)l=mid+1;else r=mid;
	return l;
}
int main(){
	n=read(),m=read();
	rep(i,1,m){
		scanf("%s%d%d",ch,&x[i],&y[i]);num[i]=y[i];
		if(ch[0]=='U')z[i]=1;else z[i]=0;
	}
	sort(num+1,num+1+m);hash[cnt=1]=num[1];
	rep(i,2,m)
		if(num[i]!=num[i-1])hash[++cnt]=num[i];
	rep(i,1,m)y[i]=find(y[i]);
	rep(i,1,m)
		if(z[i]){
			if(j=a[x[i]]){bit1.modify(j,-1);bit2.modify(j,-hash[j]);}
			a[x[i]]=y[i];
			bit1.modify(y[i],1);bit2.modify(y[i],hash[y[i]]);
		}else puts(bit2.query(y[i]-1)>=(x[i]-bit1.query(cnt)+bit1.query(y[i]-1))*hash[y[i]]?"TAK":"NIE");
	return 0;
}
本文可以转载,但必须附上原文链接,否则你会终生找不到妹子!!!欢迎关注我的CSDN: ahyjjr