树状数组板题
T1
题目描述
如题,已知一个数列,你需要进行下面两种操作:
·将某一个数加上 xxx
·求出某区间每一个数的和
输入格式
第一行包含两个正整数 n,mn,mn,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 nnn 个用空格分隔的整数,其中第 iii 个数字表示数列第 iii 项的初始值。
接下来 mmm 行每行包含 333 个整数,表示一个操作,具体如下:
x k 含义:将第 xxx 个数加上 kkk
x y 含义:输出区间 [x,y][x,y][x,y] 内每个数的和
输出格式
输出包含若干行整数,即为所有操作 222 的结果。
Code
#include <bits/stdc++.h>
using namespace std;
int a[5000000];
int c[5000000];
int n,m;
int lowbit(int x){
return x&(-x);
}
void add(int k,int x){
a[k] = x;
for(int i=1;k-lowbit(k)+i<=k;i++){
c[k]+=a[k-lowbit(k)+i];
}
return;
}
int shu(int x,int y){
int sum=0;
for(int i=y;i>0;i-=lowbit(i)){
sum+=c[i];
}
for(int i=x;i>0;i-=lowbit(i)){
sum-=c[i];
}
return sum;
}
void change(int x,int y){
for(int i=x;i<=n;i+=lowbit(i)){
c[i]+=y;
}
return;
}
int main(){
cin>>n>>m;
for(int i = 1;i <= n;i++){
int x;
cin>>x;
add(i,x);
}
while(m--){
int vis,x,y;
cin>>vis>>x>>y;
if(vis == 2){
cout<<shu(x-1,y)<<endl;
}else {
change(x,y);
}
}
return 0;
}
T2
题目描述
如题,已知一个数列,你需要进行下面两种操作:
将某区间每一个数数加上 xxx;
求出某一个数的值。
输入格式
第一行包含两个整数 NNN、MMM,分别表示该数列数字的个数和操作的总个数。
第二行包含 NNN 个用空格分隔的整数,其中第 iii 个数字表示数列第 ii i 项的初始值。
接下来 MMM 行每行包含 222 或 444个整数,表示一个操作,具体如下:
操作 111: 格式:1 x y k 含义:将区间 [x,y][x,y][x,y] 内每个数加上 kkk;
操作 222: 格式:2 x 含义:输出第 xxx 个数的值。
输出格式
输出包含若干行整数,即为所有操作 222 的结果。
Code
#include <bits/stdc++.h>
const int N = 5e6+5;
using namespace std;
int n,m;
long long c[N],shu[N];
int lowbit(int x){
return x&-x;
}
void add(int x,long long k){
while(x<=n){
shu[x]+=k;
x+=lowbit(x);
}
}
long long sum(int x){
long long ans=0;
while(x>0){
ans+=shu[x];
x-=lowbit(x);
}
return ans;
}
int main(){
cin>>n>>m;
long long last = 0,now;
for(int i=1;i<=n;i++) {
cin>>now;
add(i,now-last);
last = now;
}
int x,y,z;
long long k;
for(int i=1;i<=m;i++){
cin>>x;
if(x==1){
cin>>y>>z>>k;
add(y,k);
add(z+1,-k);
}
else if(x==2){
cin>>y;
cout<<sum(y)<<endl;
}
}
return 0;
}