BZOJ 4378: [POI2015]Logistyka 树状数组

4378: [POI2015]Logistyka

Time Limit: 20 Sec  Memory Limit: 256 MB
Submit: 528  Solved: 267
[Submit][Status][Discuss]

Description

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

Input

第一行包含两个正整数n,m(1<=n,m<=1000000)，分别表示序列长度和操作次数。
接下来m行为m个操作，其中1<=k,c<=n，0<=a<=10^9，1<=s<=10^9。

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

BZOJ4378 POI2015 Logistyka

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline ll read()
{
	ll x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=2001000;

ll n,m,val[N],opt[N],u[N],v[N],c[N],a[N],num;

ll bit[N];

map<ll,int>mp;

inline void modify(ll x,ll val)
{for(;x<=num;x+=(x&-x))bit[x]+=val;}

inline ll query(ll x)
{ll res=0;for(;x;x-=(x&-x))res+=bit[x];return res;}

inline void my(ll x,ll val)
{for(;x<=num;x+=(x&-x))c[x]+=val;}

inline ll qy(ll x)
{ll res=0;for(;x;x-=(x&-x))res+=c[x];return res;}

int main()
{
	n=read();m=read();char ch[10];
	register int i;
	for(i=1;i<=m;++i){scanf("%s",ch);opt[i]=ch[0]=='U';u[i]=read();v[i]=read();val[i]=v[i];}val[m+1]=0;
	sort(val+1,val+2+m);val[0]=-5645465;
	for(i=1;i<=m+1;++i)if(val[i]!=val[num])val[++num]=val[i];
	for(i=1;i<=num;++i)mp[val[i]]=i;
	for(i=1;i<=n;++i)my(mp[0],1);
	for(i=1;i<=m;++i)
	{
		if(opt[i])
		{
			modify(mp[a[u[i]]],-a[u[i]]);
			modify(mp[v[i]],v[i]);
			my(mp[a[u[i]]],-1);
			my(mp[v[i]],1);
			a[u[i]]=v[i];
		}
		else 
		{
			if(v[i]*u[i]<=query(mp[v[i]])+v[i]*(n-qy(mp[v[i]])))puts("TAK");
			else puts("NIE");
		}
	}
	return 0;
}
/*
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

NIE
TAK
NIE
TAK
*/