[FZYZOJ 1821] 一道果题

P1821 -- 一道果题

时间限制：1000MS

内存限制：131072KB

Description

动态单点修改 查询某段区间和。

Input Format

输入N M

第二行是N个整数 表示 初始序列.

接下来是M个询问

Q a b 表示查询[a,b]的和

M a b 表示把a位置的数替换成b

Output Format

对于每个Q 输出当前的区间和。

Sample Input

3 4
1 2 3
Q 1 2
Q 1 3
M 2 -1
Q 1 3

Sample Output

3
6
3

Hint

N,M<=100000

运行过程中所有数据均在 int / integer 范围内

【题解】

用树状数组维护即可。

#include<stdio.h>
char B[1<<15],*S=B,*T=B;
#define getc() (S==T&&(T=(S=B)+fread(B,1,1<<15,stdin),S==T)?0:*S++)
int n,m,b[100001],c[100001],a[100001];
int lowbit(int x) {return x&(-x);}
int getint() {
    int x=0,f=1; char ch=getc();
    while(ch<'0'||ch>'9') {if(ch=='-') f=-1; ch=getc();}
    while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+ch-'0'; ch=getc();}
    return x*f;
}
void modify(int x,int delta) {
    while(x<=n) {
        c[x]+=delta;
        x+=lowbit(x);
    }
} 
int cal(int x) {
    int ret=0;
    while(x!=0) {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
main() {
    n=getint();m=getint();
    for (int i=1;i<=n;++i) {
        b[i]=getint();
        a[i]=b[i]+a[i-1];
        c[i]=a[i]-a[i-lowbit(i)];
    }
    while(m--) {
        char ch=getc();
        if(ch=='Q') {
            int p,q;p=getint();q=getint();
            printf("%d\n",cal(q)-cal(p-1));
        }
        if(ch=='M') {
            int p,q;p=getint();q=getint();
            int delta=q-b[p]; b[p]=q;
            modify(p,delta);
        }
    }
}