luogu P3374 【模板】树状数组 1

题目描述

如题,已知一个数列,你需要进行下面两种操作:

1.将某一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式:

第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数,表示一个操作,具体如下:

操作1: 格式:1 x k 含义:将第x个数加上k

操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和

输出格式:

输出包含若干行整数,即为所有操作2的结果。

输入输出样例

输入样例#1:
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4
输出样例#1:
14
16

说明

时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=8,M<=10

对于70%的数据:N<=10000,M<=10000

对于100%的数据:N<=500000,M<=500000

样例说明:

故输出结果14、16

树状数组,模板题,没有什么特别的。

 1 #include<cstdio>
 2 
 3 const int N=10000001;
 4 
 5 int n,m;
 6 int a[N];
 7 
 8 int b,c,d;
 9 
10 int lowbit(int x){
11     return x&(-x);
12 }
13 
14 
15 void add(int i,int x)
16 {
17     while(i<=n)
18     {
19         a[i]+=x;
20         i+=lowbit(i);
21     }
22 }
23 
24 int query(int x)
25 {
26     int ans=0;
27     //if(x<y)swap(x,y);
28     while(x>0)
29     {
30         ans+=a[x];
31         x-=lowbit(x);
32     }
33     return ans;
34 }
35 int main()
36 {
37     scanf("%d%d",&n,&m);
38     int q;
39     for(int i=1;i<=n;i++)
40     {
41         scanf("%d",&q);
42         add(i,q);
43     }
44     for(int i=1;i<=m;i++)
45     {
46         scanf("%d%d%d",&b,&c,&d);
47         if(b==1)
48         {
49              add(c,d);
50         }
51         if(b==2)
52         {
53             printf("%d
",query(d)-query(c-1));
54         }
55     }
56     return 0;
57 }
原文地址:https://www.cnblogs.com/sssy/p/6827832.html