首页 > 代码库 > 洛谷——P3368 【模板】树状数组 2
洛谷——P3368 【模板】树状数组 2
https://www.luogu.org/problem/show?pid=3368
题目描述
如题，已知一个数列，你需要进行下面两种操作：
1.将某区间每一个数数加上x
2.求出某一个数的和
输入输出格式
输入格式：
第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数，其中第i个数字表示数列第i项的初始值。
接下来M行每行包含2或4个整数，表示一个操作，具体如下：
操作1： 格式：1 x y k 含义：将区间[x,y]内每个数加上k
操作2： 格式：2 x 含义：输出第x个数的值
输出格式：
输出包含若干行整数，即为所有操作2的结果。
输入输出样例
输入样例#1：
5 51 5 4 2 31 2 4 22 31 1 5 -11 3 5 72 4
输出样例#1：
610
说明
时空限制：1000ms,128M
数据规模：
对于30%的数据：N<=8，M<=10
对于70%的数据：N<=10000，M<=10000
对于100%的数据：N<=500000，M<=500000
样例说明：
故输出结果为6、10
1 #include <cstdio> 2 3 #define lowbit(x) (x&(-x)) 4 5 using namespace std; 6 7 const int N(500000+15); 8 int n,m,x,y,k,tr[N]; 9 int if_,ch,od;10 11 inline void read(int &x)12 {13 if_=x=0; ch=getchar();14 for(;ch<‘0‘||ch>‘9‘;ch=getchar())15 if(ch==‘-‘) if_=1;16 for(;ch>=‘0‘&&ch<=‘9‘;ch=getchar())17 x=x*10+ch-‘0‘;18 if(if_) x=(~x)+1;19 }20 21 inline void up(int i,int x)22 {23 for(;i<=n;i+=lowbit(i)) tr[i]+=x;24 }25 26 inline int query(int x)27 {28 int ret=0;29 for(;x;x-=lowbit(x)) ret+=tr[x];30 return ret;31 }32 33 int main()34 {35 read(n); read(m);36 for(int i=1;i<=n;i++)37 read(x),up(i,x-k),k=x;38 for(;m;m--)39 {40 read(od); 41 if(od==1)42 {43 read(x);read(y);read(k);44 up(x,k); up(y+1,-k);45 }46 else47 {48 read(x);49 printf("%d\n",query(x));50 }51 }52 return 0;53 }
洛谷——P3368 【模板】树状数组 2
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。