UVa12086 - Potentiometers

题目大意

给定n个数，你的任务是实现如下两种操作：

“S x y“ 把第x个数修改为y

“M x y”计算第X个数到第Y个数之和

题解

这刚好是树状数组的功能，修改元素的值和求区间和

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#define MAXN 200005
using namespace std;
int n;
int c[MAXN];
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int d)
{
    while(x<=n)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int ret=0;
    while(x>0)
    {
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
int main(void)
{
    int i,k,x,y,p,t;
    bool first_case=true;
    string s;
    p=0;
    while(scanf("%d",&n)&&n)
    {
        if (!first_case)
            cout << endl;
        first_case = false;
        printf("Case %d:\n",++p);
        memset(c,0,sizeof(c));
        for(i=1; i<=n; i++)
        {
            scanf("%d",&k);
            add(i,k);
        }
        while(cin>>s)
        {
            if(s=="END") break;
            if(s=="M")
            {
                scanf("%d%d",&x,&y);
                printf("%d\n",sum(y)-sum(x-1));
            }
            else
            {
                scanf("%d%d",&x,&y);
                t=sum(x)-sum(x-1);
                add(x,y-t);
            }
        }
    }
    return 0;
}