uestc 1073 秋实大哥与线段树 Label：线段树

秋实大哥与线段树

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 1 2 3 3 2 1 2 1 1 5 2 1 2	3 7

代码  树状数组

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define ll long long
using namespace std;
ll bit[500005],a[500005];
ll n,m;

ll sum(ll i){//求元素1~i的和 
    ll s=0;
    while(i>0){
        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=1;i<=n;i++){
        scanf("%d",&a[i]);
        add(i,a[i]);
    }
    
    scanf("%lld",&m);
    for(int i=1;i<=m;i++){
        ll x=0,b=0,c=0;
        scanf("%lld%lld%lld",&x,&b,&c);
        if(x==1){
            ll temp=c-a[b];//处理修改后的数与原数的差 
            a[b]=c; 
            add(b,temp);
        }
        else if(x==2){
            ll temp=sum(c)-sum(b-1);
            printf("%lld\n",temp);
        }
    }
    return 0;
}

用的树状数组，注意变量名要起好，像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[10000005];//记得改long long 
ll n,temp,M,m;

void query(ll s,ll t){
    ll ans=0;
    
    for(s=s+M-1,t=t+M+1;s^t^1;s>>=1,t>>=1){
        if(~s&1) ans+=tr[s^1];
        if(t&1)  ans+=tr[t^1];
    }
    
    printf("%lld\n",ans);
}

void change(ll x,ll y){
    for(tr[x+=M]+=y,x>>=1;x>0;x>>=1){
        tr[x]=tr[x<<1]+tr[x<<1|1];
    }
}

int main(){
//    freopen("01.in","r",stdin);
    scanf("%lld",&n);
    for(M=1;M<(n+2);M<<=1);//定义M大小 

    for(ll i=1;i<=n;i++){
        scanf("%lld",&temp);
        change(i,temp);
    }
    
    scanf("%lld",&m);
    
    for(ll i=1;i<=m;i++){
        ll oper,s,t;
        scanf("%lld%lld%lld",&oper,&s,&t);
        if(oper==2){
            query(s,t);
        }
        else if(oper==1){
            temp=t-tr[s+M];
            change(s,temp);
        }
    }
    return 0;
}

Line 22

x>>=1

一定要写

 

还有就是Line 30之前放到Line 36，诡异的错误