给定 n 个数组成的一个数列,规定有两种操作,一是修改某个元素,二是求子数列 [a,b]的连续和。
输入格式
第一行包含两个整数 n和 m,分别表示数的个数和操作次数。
第二行包含 n个整数,表示完整数列。
接下来 m行,每行包含三个整数 k,a,b (k=0,表示求子数列[a,b]和;k=1,表示第 a 个数加 b)。
数列从 1开始计数。
输出格式
输出若干行数字,表示 k=0时,对应的子数列 [a,b]的连续和。
数据范围
1≤n≤100000,1≤m≤100000,1≤a≤b≤n
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8
输出样例:
11
30
35
AC代码:
#include<stdio.h>
const int N=100010;
//线段树最多有4n个结点
int a[N],tree[4*N];
//用a[l]到a[r]为线段树构建节点,父节点是node
void build(int node,int l,int r)
{
if(l==r){//l,r是原数组下标,node是线段树节点号
tree[node]=a[l];
}
else {
int mid=l+((r-l)>>1);
int left_node=node<<1;
int right_node=(node<<1)|1;
build(left_node,l,mid);
build(right_node,mid+1,r);
tree[node]=tree[left_node]+tree[right_node];
}
}
//给a[idx]加上val
void add(int node,int l,int r,int idx,int val)
{
if(l==r){
tree[node]+=val;
//a[idx]+=val;
//如果加上这句,还可以实现O(1)查询某个元素
}
else {
int mid=l+((r-l)>>1);
int left_node=node<<1;
int right_node=(node<<1)|1;
//根据要修改元素下标在哪个区间范围判断要修改哪边
if(idx>=l&&idx<=mid) add(left_node,l,mid,idx,val);
else add(right_node,mid+1,r,idx,val);
//维护父节点
tree[node]=tree[left_node]+tree[right_node];
}
}
//a,b是所求连续区间左右端点,l,r是当前查询区间
int query(int node,int l,int r,int a,int b)
{
//当前查询区间不在所求范围内,不用计算
if(a>r||b<l) return 0;
//当前区间在所求区间内,或搜索到单个元素,当前节点值可用
else if(a<=l&&b>=r||l==r) return tree[node];
else {
int mid=l+((r-l)>>1);
int left_node=node<<1;
int right_node=(node<<1)|1;
int left_sum=query(left_node,l,mid,a,b);
int right_sum=query(right_node,mid+1,r,a,b);
return left_sum+right_sum;
}
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) scanf("%d",&a[i]);
//根节点,初始区间范围
build(1,1,n);
int k,x,y;
while(m--)
{
scanf("%d%d%d",&k,&x,&y);
if(k==0) printf("%d\n",query(1,1,n,x,y));
else add(1,1,n,x,y);
}
return 0;
}
//本模板记忆窍门:
//node l r必有,剩下两个参数看情况
//mid left_node right_node必求