【树状数组】校门口的树

题目描述

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

现有两个操作： 

K=1，读入l、r表示在区间[l,r]中种上一种树，每次操作种的树的种类都不同 
K=2，读入l,r表示询问l~r之间能见到多少种树。

注意：每个座位都可以重复种树。

 

输入

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

输出

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

样例输入

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

样例输出

1
2

题解

这道题因为是求某个区间内树的数量，树在区间内的具体位置是不知道的，所以用两个数组分别表示该点左侧树的数量和该点右侧树的数量，区间[a,b]内树的数量=b点左侧树的数量 – (a-1)点右侧树的数量。之所以是a-1点不是a点是因为要算上a点上的树。

#include<iostream>
#include<cstring>
using namespace std;
int n,m,sum1[100010],sum2[100010];//sum1[i]表示i左边有多少棵树，sum2[i]表示i右边有多少颗树
int lowbit(int i)//取出i的最低位1
{
	return i&(-i);
}

int getsum(int x,int c[])
{
	int ans=0;
	for(int i=x;i>0;i-=lowbit(i))
		ans+=c[i];
	return ans;
}

void update(int i,int k,int c[])//在i处加上k的单点更新
{
	while(i<=n)
	{
		c[i]+=k;
		i+=lowbit(i);
	}
}

int main()
{
	scanf("%d %d",&n,&m);
	while(m--)
	{
		int k,a,b;
		scanf("%d %d %d",&k,&a,&b);
		if(k==1)
		{
			update(a,1,sum2);
			update(b,1,sum1);
		}
		else if(k==2)
			printf("%d\n",getsum(b,sum2)-getsum(a-1,sum1));
	}
	return 0;
}