给定 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必求