富庶的建业城中,有一条格格不入的长街,名曰跳蚤街,被战争所致的孤儿,聚集于此。全国的经济都在为战争服务之时,也无人顾得了这里了。
除了两位夫人。
大乔小乔每天都会带着一些食物来到跳蚤街,分给某一位孩子。为了避免分配不均,她们时常会询问一个区域内食物的总量,然后进行调整以保证每个孩子都有足够的食物。
Input
第一行两个整数n,m,表示跳蚤街住着n户孩子,大乔小乔一共分发或询问了m次。
第二行n个整数,第i个数ai表示第i户孩子已有ai的食物。
接下来m行,每行开始先读入一个整数si,指明这是一次询问还是一次分发。
si=0,表明这是一次询问,然后读入两个整数li,ri,表示询问[li,ri]区间中的孩子们一共有多少食物。
si=1,表明这是一次分发,然后读入两个整数xi,wi,表示对第xi户孩子分发了wi的食物。
1≤n,m≤100000,0≤ai≤100000,1≤xi≤n,0≤wi≤10000,1≤li≤ri≤n
Output
有多少询问就输出多少行,每行输出一个整数,作为对该询问的回答。
Sample input and output
Sample Input | Sample Output |
---|---|
5 4 1 2 3 4 5 1 2 3 0 2 4 1 4 1 0 1 5 |
12 19 |
纯裸的线段树,没什么好说的。。关于线段树大家可以百度一下或者google一下,会有很多文章有介绍。
我这段程序是按jianshu(建树),suan(给线段树赋初值,即一开始的i孩子有ai食物),found(数的区间查找)|done(树的维护)的顺序写的。
#include <iostream> #include<stdio.h> #include<string.h> using namespace std; int q[100005],shu[400005][3]; void jianshu(int g,int h,int l) { shu[l][1]=g;shu[l][2]=h; if (g==h) {shu[l][0]=q[g];return;} jianshu(g,(g+h)/2,l*2); jianshu((g+h)/2+1,h,l*2+1); } int suan(int j) { if (shu[j][1]==shu[j][2]) return shu[j][0]; return shu[j][0]=suan(j*2)+suan(j*2+1); } int found(int l,int g,int h) { if (g==shu[l][1] && h==shu[l][2]) return shu[l][0]; int z=(shu[l][1]+shu[l][2])/2; if (g<=z && h>=z+1) return found(l*2,g,z)+found(l*2+1,z+1,h); if (h<=z) return found(l*2,g,h);else return found(l*2+1,g,h); } void done(int l,int g,int h) { shu[l][0]+=h; if (shu[l][1]==shu[l][2]) return; if (g<=(shu[l][1]+shu[l][2])/2) done(l*2,g,h);else done(l*2+1,g,h); } int main() { int n,m,j,l,g,h; cin>>n>>m; memset(q,123,sizeof(q)); memset(shu,0,sizeof(shu)); for (j=1;j<=n;j++) scanf("%d",&q[j]); jianshu(1,n,1); suan(1); for (j=1;j<=m;j++){ scanf("%d%d%d",&l,&g,&h); if (l==0) printf("%d\n",found(1,g,h)); if (l==1) done(1,g,h); } return 0; }