vijos1320 清点人数题解

最新推荐文章于 2022-01-04 20:44:35 发布

zyq_20030305

最新推荐文章于 2022-01-04 20:44:35 发布

阅读量520

点赞数

分类专栏：树状数组算法数据结构文章标签：数据结构算法树状数组

本文链接：https://blog.csdn.net/zyq_20030305/article/details/78290638

版权

算法同时被 3 个专栏收录

46 篇文章 0 订阅

订阅专栏

数据结构

12 篇文章 0 订阅

订阅专栏

树状数组

5 篇文章 0 订阅

订阅专栏

Description

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

Input

第一行两个整数n,k，表示火车共有n节车厢以及k个事件。接下来有k行，按时间先后给出k个事件，每行开头都有一个字母A,B或C，如果字母为A，接下来是一个数m，表示年级主任现在在第m节车厢；如果为B，接下来两个数m,p，表示在第m节车厢有p名学生上车；如果为C，接下来两个数m,p，表示在第m节车厢有p名学生下车。学生总人数不会超过100000。

Output

有多少个A就输出多少行，每行一个整数，回答年级主任提出的问题。

Sample Input

10 7

A 1

B 1 1

B 3 1

B 4 1

A 2

A 3

A 10

Sample Output

HINT

注意:对于30%的数据,n< =10000,k< =10000 至少有3000个A 对于100%的数据n< =500000,k< =100000. 至少有30000个A

树状数组裸题,点修改,区间查询。

#include<stdio.h>
int c[500001],n;
char a[2];
int k,m,p,i;
int query(int x)
{
	int sum=0;
	while(x)
	{
		sum+=c[x];
		x-=x&-x;
	}
	return sum;
}
void update(int x,int delta)
{
	while(x<=n)
	{
		c[x]+=delta;
		x+=x&-x;
	}
}
int main()
{
	scanf("%d%d",&n,&k);
 	for(i=1;i<=k;i++)
 	{
	 	scanf("%s",&a);
		if(a[0]=='A')
		{
			scanf("%d",&m);
			int ans=query(m);
			printf("%d\n",ans);
		}	 	
		if(a[0]=='B')
		{
			scanf("%d%d",&m,&p);
			update(m,p);
		}
		if(a[0]=='C')
		{
			scanf("%d%d",&m,&p);
			update(m,-p);
		}
 	}
}

不用解释了,不会树状数组可以看我的另一篇博客。