线段树|计蒜客:斑点蛇-区间和-单点更新
斑点蛇
有一种神奇斑点蛇,蛇如其名,全身都是斑点,斑点数量可以任意改变。
有一天,蒜头君十分的无聊,开始数蛇上的斑点。假设这条蛇的长度是Ncm,蒜头君已经数完开始时蛇身的第icm上有ai个斑点。
现在蒜头君想知道这条斑点蛇的任意区间的蛇身上一共有多少个斑点。这好像是一个很容易的事情,但是这条蛇好像是和蒜头君过不去,总是刻意的改变蛇身上的斑点数量。
于是,蒜头君受不了了,加上蒜头君有密集型恐惧症。聪明又能干的你能帮帮他吗?
输入格式
第一行一个正整数N(N≤50000)表示这条斑点蛇长度为N厘米,接下来有N个正整数,第个正整数a;代表第i个斑点蛇第i厘米开始时有a;个斑点(1≤ai≤50)。
接下来每行有一条命令,命令有4种形式:
(1)Addij,i和j为正整数,表示第i厘米增加j个斑点(j不超过30);
(2)Subij,i和i为正整数,表示第i厘米减少j个斑点(不超过30);
(3)Queryij,i和j为正整数,i≤j,表示询问第i到第j厘米的斑点总数(包括第i厘米和第j厘米);
(4)End 表示结束,这条命令在每组数据最后出现;
输出格式
对于每个Query询问,输出一个整数并回车,表示询问的段中的总斑点数,这个数保证在int 范围内。
样例输入
10
1 2 3 4 5 6 7 8 9 1 0
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End
样例输出
6
33
59
思路:线段树模板,维护区间和
代码:
#include<bits/stdc++.h>
using namespace std;
const int max_n = 50005;
int n;
int s[4 * max_n];
//父节点的值 等于:合并左右子节点的值
void up(int p){
s[p] = s[p<<1] + s[(p<<1) + 1];
}
//p当前结点 L边界 r右边界 x要更新的下标 v要更新为的值
void modify(int p,int l,int r,int x,int v){
if(l == r){
s[p] += v;
return;
}
int mid = l + (r - l)/2;
if(x <= mid){
modify(p<<1, l, mid, x, v); //左子树 左区间更新
}else{
modify((p<<1) + 1, mid + 1, r, x, v);
}
up(p); //父节点合并 两个子节点
}
//查询
int query(int p,int l,int r,int x,int y){
if(x <=l && r <=y){
return s[p];
}
int mid = l + (r - l)/2,res = 0;
if(x<=mid){
res += query(p<<1,l,mid,x,y);
}
if(y>mid){
res += query((p<<1) +1,mid+1,r,x,y);
}
return res;
}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
int d;
scanf("%d",&d);
modify(1,1,n,i,d);
}
string ins;
int i,j;
while(cin>>ins){
if(ins == "End") break;
scanf("%d%d",&i,&j);
if(ins == "Query"){
printf("%d\n",query(1,1,n,i,j));
}else if(ins == "Add"){
modify(1,1,n,i,j);
}else{
j = -j;
modify(1,1,n,i,j);
}
}
return 0;
}