清点人数

https://loj.ac/problem/10116

题目描述

  有\(n\)节车厢，有k个操作，分为三种：\(①\)询问前m节车厢中的总人数；\(②\)第m节车厢增加x人；\(③\)第m节车厢减少x人。

思路

  维护前缀和，树状数组的模板题，直接套板子即可。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=5e5+10;

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

int c[N],n;
int lowbit(int x)
{
	return x&-x;
}
void add(int x,int w)
{
	while(x<=n)
	{
		c[x]+=w;
		x+=lowbit(x);
	}
}
int query(int x)
{
	int res=0;
	while(x)
	{
		res+=c[x];
		x-=lowbit(x);
	}
	return res;
}
int main() 
{
	int k;
	n=read();k=read();
	while(k--)
	{
		char ch;
		scanf(" %c",&ch);
		if(ch=='A')
		{
			int m=read();
			write(query(m));
			putchar('\n');
//			printf("%d\n",query(m));
		}
		else
		{
			int m=read(),p=read();
			if(ch=='B')add(m,p);
			else add(m,-p);
		}
	}
}