Hdu1166_敌兵布阵——线段树or树状数组

树状数组，插点问段

#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=50000+5;
int c[MAXN];
int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int temp=0;
    while(x>0){
        temp+=c[x];
        x-=lowbit(x);
    }
    return temp;
}
void update(int x,int num,int n)
{
    while(x<=n){
        c[x]+=num;
        x+=lowbit(x);
    }
}
int main()
{
    int test;
    scanf("%d",&test);
    for(int cas=1;cas<=test;cas++){
        printf("Case %d:
",cas);
        int n;
        scanf("%d",&n);
        memset(c,0,sizeof(c));
        int a;
        for(int i=1;i<=n;i++){
            scanf("%d",&a);
            update(i,a,n);
        }
        char s[10];
        int u,v;
        loop:scanf("%s",&s);
        if(s[0]=='A'){
            scanf("%d%d",&u,&v);
            update(u,v,n);
            goto loop;
        }
        else if(s[0]=='S'){
            scanf("%d%d",&u,&v);
            update(u,-v,n);
            goto loop;
        }
        else if(s[0]=='Q'){
            scanf("%d%d",&u,&v);
            printf("%d
",sum(v)-sum(u-1));
            goto loop;
        }
        else
            continue;
    }
    return 0;
}

线段树，单点修改

#include<cstdio>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=50010;
int sum[maxn<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r){
        scanf("%d",&sum[rt]);
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}
void update(int p,int add,int l,int r,int rt)
{
    if(l==r){
        sum[rt]+=add;
        return ;
    }
    int m=(l+r)>>1;
    if(p<=m)
        update(p,add,lson);
    else
        update(p,add,rson);
    pushup(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return sum[rt];
    int m=(l+r)>>1;
    int ret=0;
    if(L<=m)
        ret+=query(L,R,lson);
    if(R>m)
        ret+=query(L,R,rson);
    return ret;
}
int main()
{
    int test;
    scanf("%d",&test);
    for(int cas=1;cas<=test;cas++){
        printf("Case %d:
",cas);
        int n;
        scanf("%d",&n);
        build(1,n,1);
        char s[10];
        int i,j;
        while(scanf("%s",&s)){
            if(s[0]=='A'){
                scanf("%d%d",&i,&j);
                update(i,j,1,n,1);
            }
            else if(s[0]=='S'){
                scanf("%d%d",&i,&j);
                update(i,-j,1,n,1);
            }
            else if(s[0]=='Q'){
                scanf("%d%d",&i,&j);
                printf("%d
",query(i,j,1,n,1));
            }
            else if(s[0]=='E')
                break;
        }
    }
    return 0;
}