[模板] zkw线段树

zkw线段树

code1简单版本 code2差分版本(暂无)

code1:(有注释)

//By Menteur_Hxy
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAX=100010;
const int INF=0x3f3f3f3f;
int n,m,M;
long long rd() { //快读(*/ω＼*)
    long long x=0,fla=1;
    char c=' ';
    while(c>'9'|| c<'0') {if(c=='-') fla=-fla; c=getchar();}
    while(c<='9'&&c>='0') x=x*10+c-'0',c=getchar();
    return x*fla;
}
struct zkw_Segment_Tree { //无需差分 支持单点修改和区间查询 （求和 和 最值 都可以）
    long long a[MAX<<1];
    void build() {
        for(M=1;M<n;M<<=1); //强行开够（大于n） 方便二进制访问叶节点
        for(int i=M+1;i<=M+n;i++) a[i]=rd();
        /*这里要解释下。。
            理论上从M开始都是叶节点那为什么不从M开始而是M+1呢
            因为在区间查询时最大是[1,n]而这个线段树区间查询需要变成开区间
            即变为(0,n+1) 所以当然要给0的位置留个空 区间查询才能 
        */
        for(int i=M-1;i;i--) a[i]=a[i<<1]+a[i<<1|1];
    }
    void point_change(long long x,long long k) {
        a[x=x+M]+=k;
        while(x) a[x>>=1]=a[x<<1]+a[x<<1|1];
    }
    long long query_interval (long long l,long long r) {
        long long ans=0;
        for(l=l+M-1,r=r+M+1;l^r^1;l>>=1,r>>=1) {
            if(~l&1) ans+=a[l^1];
            if(r&1) ans+=a[r^1];
        }
        return ans;
    }
}tr;
int main() {
    n=rd(),m=rd();
    tr.build();
    for(int i=1;i<=m;i++) {
        int opt=rd();
        if(opt==1) {
            int x=rd(),k=rd();
            tr.point_change(x,k);
        }
        else {
            int l=rd(),r=rd();
            if(l>r) swap(l,r);
            printf("%d",tr.query_interval(l,r));
        }
    }
    return 0;
}

code1:(无注释)

//By Menteur_Hxy
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAX=100010;
const int INF=0x3f3f3f3f;
int n,m,M;
long long rd() {
    long long x=0,fla=1;
    char c=' ';
    while(c>'9'|| c<'0') {if(c=='-') fla=-fla; c=getchar();}
    while(c<='9'&&c>='0') x=x*10+c-'0',c=getchar();
    return x*fla;
}
struct zkw_Segment_Tree {
    long long a[MAX<<1];
    void build() {
        for(M=1;M<n;M<<=1);
        for(int i=M+1;i<=M+n;i++) a[i]=rd();
        for(int i=M-1;i;i--) a[i]=a[i<<1]+a[i<<1|1];
    }
    void point_change(long long x,long long k) {
        a[x=x+M]+=k;
        while(x) a[x>>=1]=a[x<<1]+a[x<<1|1];
    }
    long long query_interval (long long l,long long r) {
        long long ans=0;
        for(l=l+M-1,r=r+M+1;l^r^1;l>>=1,r>>=1) {
            if(~l&1) ans+=a[l^1];
            if(r&1) ans+=a[r^1];
        }
        return ans;
    }
}tr;
int main() {
    n=rd(),m=rd();
    tr.build();
    for(int i=1;i<=m;i++) {
        int opt=rd();
        if(opt==1) {
            int x=rd(),k=rd();
            tr.point_change(x,k);
        }
        else {
            int l=rd(),r=rd();
            if(l>r) swap(l,r);
            printf("%d",tr.query_interval(l,r));
        }
    }
    return 0;
}