【BZOJ4378】[POI2015]Logistyka 树状数组

【BZOJ4378】[POI2015]Logistyka

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

题解:我们考虑什么情况下询问有解。若一个数>s,那么我们肯定贪心的每次都让它-1,但是它最多只能取s次,所以它跟s没什么区别;若一个数≤s,那我们贪心的将它取到0,它对总和的贡献就是它本身。所以综上所述,有解的条件就是 ∑min(v[i],s) (1≤i≤n)≥c*s。我们只需要维护两个树状数组,一个记录>s的数的个数,一个记录≤s的数的总和,然后只要 个数*s+总和≥c*s就行了

需要离散化,别忘开long long

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=1000010;
int n,m,nm;
char str[5];
struct node
{
	int org;
	ll num;
}p[maxn<<1];
int qa[maxn],qc[maxn];
ll ref[maxn<<1],s1[maxn],s2[maxn],v[maxn],qb[maxn];
int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<'0'||gc>'9')	{if(gc=='-')f=-f;	gc=getchar();}
	while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();
	return ret*f;
}
void up1(int x,ll val)
{
	for(int i=x;i<=nm;i+=i&-i)	s1[i]+=val;
}
void up2(int x,ll val)
{
	for(int i=x;i<=nm;i+=i&-i)	s2[i]+=val;
}
ll q1(int x)
{
	int i=x;
	ll ret=0;
	for(i=x;i;i-=i&-i)	ret+=s1[i];
	return ret;
}
ll q2(int x)
{
	int i=x;
	ll ret=0;
	for(i=x;i;i-=i&-i)	ret+=s2[i];
	return ret;
}
bool cmp(node a,node b)
{
	return a.num<b.num;
}
int main()
{
	n=rd(),m=rd();
	int i,j,a,b;
	for(i=1;i<=m;i++)
	{
		scanf("%s",str),qa[i]=rd(),p[i].num=rd(),p[i].org=i;
		if(str[0]=='U')	qc[i]=0;
		else	qc[i]=1;
	}
	sort(p+1,p+m+1,cmp);
	ref[1]=0,nm=1;
	for(i=1;i<=m;i++)
	{
		if(p[i].num>ref[nm])	ref[++nm]=p[i].num;
		qb[p[i].org]=nm;
	}
	for(i=1;i<=n;i++)	up1(1,1),v[i]=1;
	for(i=1;i<=m;i++)
	{
		if(!qc[i])
		{
			up1(v[qa[i]],-1),up2(v[qa[i]],-ref[v[qa[i]]]),v[qa[i]]=qb[i];
			up1(qb[i],1),up2(qb[i],ref[qb[i]]);
		}
		else
		{
			if((n-q1(qb[i]))*ref[qb[i]]+q2(qb[i])>=qa[i]*ref[qb[i]])	printf("TAK\n");
			else	printf("NIE\n");
		}
	}
	return 0;
}
posted @ 2017-05-10 11:46  CQzhangyu  阅读(282)  评论(0编辑  收藏  举报