树状数组 校门外的树
时间限制: 1 Sec 内存限制: 128 MB
题目描述
校门外有很多树,有苹果树,香蕉树,有会扔石头的,有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树,保证任一时刻不会出现两段相同种类的树,现有两个操作:
K=1,K=1,读入l、r表示在区间[l,r]中种上一种树,每次操作种的树的种类都不同
K=2,读入l,r表示询问l~r之间能见到多少种树
(l,r>0)
输入
第一行n,m表示道路总长为n,共有m个操作
接下来m行为m个操作
输出
第一行n,m表示道路总长为n,共有m个操作
接下来m行为m个操作
样例输入
5 4
1 1 3
2 2 5
1 2 4
2 3 5
样例输出
1
2
就是树状数组吗,直接水过。
这个题要用两个树状树组。但总的来说,它的思路是用前缀和相减的方式,维护出答案。
它会动态地插入一些区间,在这个区间中,就多了一种树。于是我们可以用两个数组 l[ ] 和 r[ ],来分别维护前 i 个点,有多少个区间左界,有多少个区间右界。询问区间[a,b]时,只要写出l[b]-r[a-1]就可以了。因为l[b]可以表示出在[1,b]有多少个区间开始,而r[a-1]又可以表示出在a之前有多少个区间结束,这样一减就是答案。
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int t1[50010],t2[50010];
int m,n,k,l,r;
int lowbit(int x){
return x&(-x);
}
void add_t1(int x,int y){
while(x<=n){
t1[x]+=y;
x+=lowbit(x);
}
}
void add_t2(int x,int y){
while(x<=n){
t2[x]+=y;
x+=lowbit(x);
}
}
int sum_t1(int x){
int ans=0;
while(x>0){
ans+=t1[x];
x-=lowbit(x);
}
return ans;
}
int sum_t2(int x){
int ans=0;
while(x>0){
ans+=t2[x];
x-=lowbit(x);
}
return ans;
}
int main()
{
int i;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++){
scanf("%d%d%d",&k,&l,&r);
if(k==1)add_t1(l,1),add_t2(r,1);
else printf("%d\n",sum_t1(r)-sum_t2(l-1));
}
}