#10116 「一本通 4.1 练习 1」清点人数

Posted 2020-11-11 fdezlsq

1538：清点人数

时间限制: 200 ms         内存限制: 524288 KB
提交数: 920     通过数: 204 

【题目描述】

NK 中学组织同学们去五云山寨参加社会实践活动，按惯例要乘坐火车去。由于 NK 中学的学生很多，在火车开之前必须清点好人数。

初始时，火车上没有学生。当同学们开始上火车时，年级主任从第一节车厢出发走到最后一节车厢，每节车厢随时都有可能有同学上下。年级主任走到第 mm 节车厢时，他想知道前 mm 节车厢上一共有多少学生，但是他没有调头往回走的习惯。也就是说每次当他提问时，mm 总会比前一次大。

【输入】

第一行两个整数 n,kn,k，表示火车共有 nn 节车厢以及 kk 个事件。

接下来有 kk 行，按时间先后给出 kk 个事件，每行开头都有一个字母 AA，BB 或 CC。

如果字母为 AA，接下来是一个数 mm，表示年级主任现在在第 mm 节车厢；

如果字母为 BB，接下来是两个数 m,pm,p，表示在第 mm 节车厢有 pp 名学生上车；

如果字母为 CC，接下来是两个数 m,pm,p，表示在第 mm 节车厢有 pp 名学生下车。

学生总人数不会超过 105105 。

【输出】

对于每个 AA ，输出一行，一个整数，表示年级主任的问题的答案。

【输入样例】

10 7
A 1
B 1 1
B 3 1
B 4 1
A 2
A 3
A 10

【输出样例】

0
1
2
3

【提示】

数据范围与提示：

对于 30% 的数据，1≤n,k≤1041≤n,k≤104 ，至少有 30003000 个 AA；

对于 100% 的数据，1≤n≤5×105,1≤k≤1051≤n≤5×105,1≤k≤105 ，至少有 3×1043×104 个 AA。

【来源】

无

 

#include<bits/stdc++.h>
using namespace std;
int a[500005];
int n,k;
int Add(int x,int p){
	for(;x<=n;x+=x&(-x))a[x]+=p;
}
int Que(int x){
	int res=0;
	for(;x>0;x-=x&(-x))res+=a[x];
	return res;
}
int main(){
	//freopen("test.in","r",stdin);
	scanf("%d%d",&n,&k);
	while(k--){
		char op;int m,p;
		for(op=getchar();op<‘A‘||op>‘C‘;op=getchar());
		if(op==‘A‘){
			scanf("%d",&m);
			printf("%d\n",Que(m));
		}
		else{
			scanf("%d%d",&m,&p);
			if(op==‘C‘)p=-p;
			Add(m,p);
		}
	}
	return 0;
}