树状数组模板题
#include<bits/stdc++.h> using namespace std; #define ll long long int x[100005]={0}; int y[100005]={0}; int n,m; int er(int r)//返回二进制里的最后一个1 { return r&(-r); } void add(int r,int k)//树状数组的建立及更改 { for(int i=r;i<=n;i+=er(i)) { y[i]+=k; } } int sum(int r)//求和 { ll sum1=0; for(int i=r;i>=1;i-=er(i)) { sum1+=y[i]; } return sum1; } int main() { cin>>n>>m; for(int i=1;i<=n;i++) { cin>>x[i]; } for(int i=1;i<=n;i++) { add(i,x[i]); } while(m--) { int a,b,c; cin>>a>>b>>c; if(a==0) { cout<<sum(c)-sum(b-1)<<endl; } else { add(b,c); } } }
#include using namespace std; #define ll long long int x[100005]={0}; int y[100005]={0}; int n,m; int er(int r)//返回二进制里的最后一个1 { return r&(-r); } void add(int r,int k)//树状数组的建立及更改 { for(int i=r;i<=n;i+=er(i)) { y[i]+=k; } } int sum(int r)//求和 { ll sum1=0; for(int i=r;i>=1;i-=er(i)) { sum1+=y[i]; } return sum1; } int main() { cin>>n>>m; for(int i=1;i<=n;i++) { cin>>x[i]; } for(int i=1;i<=n;i++) { add(i,x[i]); } while(m--) { int a,b,c; cin>>a>>b>>c; if(a==0) { cout<<sum(c)-sum(b-1)<<endl; } else { add(b,c); } } }
给定 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