校门外的树
Description
校门外有很多树,有苹果树,香蕉树,有会扔石头的,有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树,保证任一时刻不会出现两段相同种类的树,现有两个操作:
K=1,读入l,r表示在l~r之间种上的一种树
K=2,读入l,r表示询问l~r之间能见到多少种树
(l,r>0)
Input
第一行n,m表示道路总长为n,共有m个操作
接下来m行为m个操作
Output
对于每个k=2输出一个答案
Sample Input
5 4
1 1 3
2 2 5
1 2 4
2 3 5
Sample Output
1
2
Hint
范围:20%的数据保证,n,m<=100
60%的数据保证,n <=1000,m<=50000
100%的数据保证,n,m<=50000
解题思路
这题就是树状数组的模板
因为树状数组本质就是前缀和,我们可以用差分优化
AC代码
#include<iostream>
#include<cstdio>
using namespace std;
int n,m,k,x,y,x1[50005],y1[50005];
int lowbit(int x){
return x&(-x);}
void ggx(int x,int y)//更改
{
for(;x<=n;x+=lowbit(x))x1[x]+=y;
}
void ggy(int x,int y)//更改
{
for(;x<=n;x+=lowbit(x))y1[x]+=y;
}
int sumx(int x)//求和
{
int sum=0;
for(;x;x-=lowbit(x))sum+=x1[x];
return sum;
}
int sumy(int x)//求和
{
int sum=0;
for(;x;x-=lowbit(x))sum+=y1[x];
return sum;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&k,&x,&y);
if(k==1)//更改
{
ggx(x,1);
ggy(y+1,1);
}
else printf("%d\n",sumx(y)-sumy(x));//差分
}
}