清点人数 树状数组模板题

1267. 清点人数 - AcWing题库

求车厢有多少个人就是求区间内的元素个数。可以转化为求区间前缀和。即刚开始所有车厢都是0人（树状数组初始化为0），有多少个人上车，就加上上车人数（前缀和），有多少人下车就加上（-下车人数）即可，此时前缀和的值就是车厢内的学生的人数。

某节车厢上车下车就是对某个点进行修改（加上某个数，减去某个数）

求前缀和 和 单点修改，经典的树状数组解法

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
int tr[N];
int n,k;


int lowbit(int x)
{
  return x&-x;	
} 

//求一个[0,X]的前缀和，即[0,X]的车厢的学生人数 
int query(int x)
{ 
  int res=0;
  for(int i=x;i;i-=lowbit(i)) res+=tr[i];	
 
return res;
}


//对一个点x加上一个值c
int add(int x,int c)
{
	for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
}




int main()
{
	cin>>n>>k;
	
      while(k--)
      {
		char op;
		cin>>op;
		
		int a,b;
		if(op=='A')
		{   
			cin>>a;
			
			//求一下区间和
		
		cout<<query(a)<<endl; 
		}
		else if(op=='B')  //有学生上车，我们对该区间增加操作 
		{
			cin>>a>>b;
			add(a,b);
		}
		else if(op=='C')            //有学生下车，我们对该区间减去某些值 
		{
			cin>>a>>b;
			add(a,-b);
		}
	}

return 0;	
}