树状数组 - 校门外的树 - 一本通 4.1 例 3

题目描述

校门外有很多树，学校决定在某个时刻在某一段种上一种树，保证任一时刻不会出现两段相同种类的树，现有两种操作：

K=1K=1，读入 l,rl,r 表示在 ll 到 rr 之间种上一种树，每次操作种的树的种类都不同；
K=2K=2，读入 l,rl,r 表示询问 ll 到 rr 之间有多少种树。
注意：每个位置都可以重复种树。

输入格式

第一行 n,mn,m 表示道路总长为 nn，共有 mm 个操作；
接下来 mm 行为 mm 个操作。

输出格式

对于每个 k=2k=2 输出一个答案。

样例

Input

5 4
1 1 3
2 2 5
1 2 4
2 3 5

Output

1
2

数据范围与提示

对于 20% 的数据，1≤n,m≤100；
对于 60% 的数据，1≤n≤10^3 ,1≤m≤5×10^4；
对于 100% 的数据，1≤n,m≤5×10^4，保证 l,r>0。

解题思路

要求一个区间内树的种类，令左右端点分别建立a,b树状数组，之后查询时，让 r 从a中查询减去 l-1 从b查询 即可查询出值。

代码如下

#include<stdio.h>
const int N=5e4+10;
#define low(a) a&(-a)
int a[N],b[N],n;
void add(int *c,int x)
{
	while(x<=n)
	{
		c[x]+=1;
		x+=low(x);
	}
}
int query(int *c,int x)
{
	int ans=0;
	while(x>0)
	{
		ans+=c[x];
		x-=low(x);
	}
	return ans;
}
int main()
{
	int m,k,l,r;
	scanf("%d %d",&n,&m);
	while(m--)
	{
		scanf("%d %d %d",&k,&l,&r);
		if(k==1)
		{
			add(a,l);
			add(b,r);
		}
		else
			printf("%d\n",query(a,r)-query(b,l-1));
	}
}