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;
}