UVALive 2191 - Potentiometers(BIT)

题目链接 https://cn.vjudge.net/problem/UVALive-2191

【题意】
给定n个整数的数组,实现以下两种操作
S x y 把第x个数改成y(1<=x<=n, 0<=y<=1000)
M x y 计算区间[x,y]的元素之和 (1<=x<=y<=n)

【输入格式】
最多两组输入数据,每组数据第一行为整数n(1<=n<=200000)以下n行每行有一个不超过1000的非负整数,接下来包含不超过200000个操作,以END结束

【输出格式】
对每组数据,输出测试编号和每个M操作的答案

【思路】
用树状数组做优化即可,也可以用线段树单点更新,但树状数组更快一些.用一个数组维护当前的元素值,要修改成的那个值减去当前的元素值就是树状数组应该加上去的值

#include<bits/stdc++.h>
using namespace std;

const int maxn=200050;

int bit[maxn],n;
int a[maxn];
char op[10];

int sum(int i){
    int ans=0;
    while(i>0){
        ans+=bit[i];
        i-=i&-i;
    }
    return ans;
}

void add(int i,int x){
    while(i<=n){
        bit[i]+=x;
        i+=i&-i;
    }
}

int main(){
    int kase=0;
    while(scanf("%d",&n)==1 && n>0){
        memset(bit,0,sizeof(bit));
        for(int i=1;i<=n;++i){
            scanf("%d",&a[i]);
            add(i,a[i]);
        }
        if(kase) puts("");
        printf("Case %d:\n",++kase);
        while(scanf("%s",op)==1){
            if(op[0]=='E') break;
            if(op[0]=='M'){
                int x,y;
                scanf("%d%d",&x,&y);
                int ans=sum(y)-sum(x-1);
                printf("%d\n",ans);
            }
            else{
                int x,y;
                scanf("%d%d",&x,&y);
                add(x,y-a[x]);
                a[x]=y;
            }
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xiao_k666/article/details/80697884