【题目描述】
校门外有很多树,有苹果树,香蕉树,有会扔石头的,有可以吃掉补充体力的……
如今学校决定在某个时刻在某一段种上一种树,保证任一时刻不会出现两段相同种类的树,现有两个操作:
K=1,读入l,r表示在l~r之间种上的一种树
K=2,读入l,r表示询问l~r之间能见到多少种树
(l,r>0)
【输入格式】
第一行n,m表示道路总长为n,共有m个操作
接下来m行为m个操作
【输出格式】
对于每个k=2输出一个答案
【样例输入】
5 4
1 1 3
2 2 5
1 2 4
2 3 5
【样例输出】
1
2
【数据范围】
n,m<=50000
【分析】
乍一看不难,仔细一看还不如乍一看。
此题完全是模板型树状数组,开两个即可。
先给出模板树状数组:
#include<bits/stdc++.h>
using namespace std;
long long a[500010],c[500010];
int n,m,x,y,k,t;
long long lowbit(long long x)
{
return x&-x;
}
long long sum(long long x)
{
long long ret=0;
while (x>0) {
ret+=c[x];
x-=lowbit(x);
}
return ret;
}
void add(long long x,long long d)
{
while (x<=n) {
c[x]+=d;
x+=lowbit(x);
}
}
int main()
{
scanf("%d%d",&n,&m);
memset(a,0,sizeof(a));
for (int i=1;i<=n;i++) { scanf("%d",&x);add(i,x); }
for (int i=1;i<=m;i++) {
scanf("%d%d%d",&k,&x,&y);
if (k==1) add(x,y); else printf("%d\n",sum(y)-sum(x-1));
}
return 0;
}
再给出此题代码:
#include<bits/stdc++.h>
using namespace std;
int l[50005],r[50005];
int n,m;
int lowbit(int x){return x&-x;}
void add(int x,int d,int c[]){
while(x<=n){
c[x]+=d;
x+=lowbit(x);
}
}
int sum(int x,int c[]){
int s=0;
while(x>0){
s+=c[x];
x-=lowbit(x);
}
return s;
}
int main(){
int k,x,y;
cin>>n>>m;
for(int i=1;i<=m;i++){
scanf("%d%d%d",&k,&x,&y);
if (k==1) add(x,1,l),add(y,1,r); else printf("%d\n",sum(y,l)-sum(x-1,r));
}
}
对比一下,不难发现仅仅是加了个数组而已(当然add和sum操作也要更改)。
线段树神马的都是浮云!树状数组才是正解!
(500+KB和1500+KB对比,你选哪个呢)