二刷算法基础题DAY9

树状数组典型应用

1264. 动态求连续区间和
      题目链接

#include<iostream>
using namespace std;

const int N=100000+10;

int a[N];
int c[N];
int n,q;

int lowbit(int x){
    return x&(-x);
}
int sum(int l){
    int sum=0;
    for(int i=l;i>=1;i-=lowbit(i)){
        sum+=c[i];
    }
    return sum;
}
void add(int x, int key){
    for(int i=x;i<=n;i+=lowbit(i)){
        c[i]+=key;
    }
}
int main(){
    cin>>n>>q;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    //求C[x]
    for(int i=1;i<=n;i++){
        add(i,a[i]);
    }
    while(q--){
        int type,l,r;
        cin>>type>>l>>r;
        if(type){   //加上某个数
            add(l,r);
        }else{  //求前缀和
            cout<<sum(r)-sum(l-1)<<endl;
        }
    }
    return 0;
}

线段树解决

#include<iostream>

using namespace std;

const int N=100000 +10;

int n,m;
int W[N];

struct Node{
    int l,r;
    int sum;
}tr[N*4];


void pushup(int u){
    tr[u].sum=tr[u<<1].sum + tr[u<<1 | 1].sum;
}

void build(int u, int l, int r){
    if(l == r)   tr[u]={l,r,W[r]};
    else{
        tr[u]={l,r};
        //除以 2 下取整
        int mid= (l + r) >> 1;
        
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        pushup(u);
        
    }
}

int query(int u, int l, int r){
    if(tr[u].l>=l && tr[u].r<=r){
        return tr[u].sum;
    }else{
        int mid=(tr[u].l+tr[u].r)>>1;
        int sum=0;
        if(l<=mid)
            sum=query(u<<1,l,r);
        if(r>mid)
            sum+=query(u<<1|1,l,r);
        return sum;
    }
    
}

void modify(int u, int x ,int v){
    if(tr[u].l==tr[u].r)    tr[u].sum+=v;
    else{
        int mid=((tr[u].l+tr[u].r)>>1);
        if(x<=mid) modify(u<<1,x,v);
        else{
            modify(u<<1|1,x,v);
        }
        pushup(u);
    }
}


int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>W[i];
    }
    
    build(1,1,n);
    int k,a,b;
    while(m--){
        cin>>k>>a>>b;
        if(k==0)
            cout<<query(1,a,b)<<endl;
        else
            modify(1,a,b);
    }
    
    return 0;
}