树状数组板题

T1

洛谷P3374 树状数组2

题目描述

如题，已知一个数列，你需要进行下面两种操作：
·将某一个数加上 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

洛谷P3368 树桩数组2

题目描述

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

将某区间每一个数数加上 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;
}