秋实大哥与线段树
Time Limit: 3000/1000MS (Java/Others) Memory Limit: 65535/65535KB (Java/Others)
“学习本无底,前进莫徬徨。” 秋实大哥对一旁玩手机的学弟说道。
秋实大哥是一个爱学习的人,今天他刚刚学习了线段树这个数据结构。
为了检验自己的掌握程度,秋实大哥给自己出了一个题,同时邀请大家一起来作。
秋实大哥的题目要求你维护一个序列,支持两种操作:一种是修改某一个元素的值;一种是询问一段区间的和。
Input
第一行包含一个整数nn,表示序列的长度。
接下来一行包含nn个整数aiai,表示序列初始的元素。
接下来一行包含一个整数mm,表示操作数。
接下来mm行,每行是以下两种操作之一:
1 x v : 表示将第x个元素的值改为v
2 l r : 表示询问[l,r]这个区间的元素和
1≤n,m,v,ai≤1000001≤n,m,v,ai≤100000,1≤l≤r≤n1≤l≤r≤n。
Output
对于每一个22 ll rr操作,输出一个整数占一行,表示对应的答案。
Sample input and output
Sample Input | Sample Output |
---|---|
3 |
3 |
代码 树状数组
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
ll bit[],a[];
ll n,m; ll sum(ll i){//求元素1~i的和
ll s=;
while(i>){
s+=bit[i];
i-=i&(-i);
}
return s;
} void add(ll i,ll x){//将元素i加上x
while(i<=n){
bit[i]+=x;
i+=i&(-i);
}
} int main(){ scanf("%lld",&n);
for(int i=;i<=n;i++){
scanf("%d",&a[i]);
add(i,a[i]);
} scanf("%lld",&m);
for(int i=;i<=m;i++){
ll x=,b=,c=;
scanf("%lld%lld%lld",&x,&b,&c);
if(x==){
ll temp=c-a[b];//处理修改后的数与原数的差
a[b]=c;
add(b,temp);
}
else if(x==){
ll temp=sum(c)-sum(b-);
printf("%lld\n",temp);
}
}
return ;
}用的树状数组,注意变量名要起好,像Line41我就直接写成了
add(x,temp);果断WA
还有Line40,没写,思考一下就知道,可能对一个值多次修改,然后Orz
代码 zkw线段树
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
ll tr[];//记得改long long
ll n,temp,M,m; void query(ll s,ll t){
ll ans=; for(s=s+M-,t=t+M+;s^t^;s>>=,t>>=){
if(~s&) ans+=tr[s^];
if(t&) ans+=tr[t^];
} printf("%lld\n",ans);
} void change(ll x,ll y){
for(tr[x+=M]+=y,x>>=;x>;x>>=){
tr[x]=tr[x<<]+tr[x<<|];
}
} int main(){
// freopen("01.in","r",stdin);
scanf("%lld",&n);
for(M=;M<(n+);M<<=);//定义M大小 for(ll i=;i<=n;i++){
scanf("%lld",&temp);
change(i,temp);
} scanf("%lld",&m); for(ll i=;i<=m;i++){
ll oper,s,t;
scanf("%lld%lld%lld",&oper,&s,&t);
if(oper==){
query(s,t);
}
else if(oper==){
temp=t-tr[s+M];
change(s,temp);
}
}
return ;
}Line 22
x>>=1一定要写
还有就是Line 30之前放到Line 36,诡异的错误