树状数组

注意：树状数组的下标要求>=1 

int tree[N];
int lowbit(int x){
	return x&-x;
} 
//a[x]+c 
int add(int x,int c){//修改往后
	for(int i=x;i<=n;i+=lowbit(i)){
		tree[i]+=c;
	} 
}
//a[1~x]的和 
int sum(int x){//计算往前
	int res=0;
	for(int i=x;i;i-=lowbit(i)){
		res+=tree[i];
	} 
	return res;
}

---

一个简单的整数问题（区间修改，单点查询）

242. 一个简单的整数问题 - AcWing题库

用树状数组维护差分

//初始化 
for(int i=1;i<=n;i++){
	cin>>w[i];
	add(i,w[i]-w[i-1]);//差分 
}

//[l,r] +d
add(l,d);add(r+1,-d);

//第x个数 
sum(x); 

---

谜一样的牛

244. 谜一样的牛 - AcWing题库

题意：n头牛的身高分别为1~n，根据每头牛前面比它矮的牛的个数，求出每头牛的身高

思路：用树状数组维护每个数是否被用过，初始时add(i,1)

           从后往前遍历，假设第i头牛前面有x头比他矮，二分答案，对于身高mid，如果[1,mid]中有x+1头牛不大于mid，则说明该头牛身高为mid        

           确定身高mid后，将其在树状数组中删除 add(mid,-1)

#include<bits/stdc++.h>
using namespace std; 
const int N=1e5+5;
int n;
int w[N];
int tree[N];
int lowbit(int x){
	return x&-x;
} 
int add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i)){
		tree[i]+=c;
	} 
}
int sum(int x){
	int res=0;
	for(int i=x;i;i-=lowbit(i)){
		res+=tree[i];
	} 
	return res;
}
int res[N];
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		if(i>1) cin>>w[i];
		add(i,1);//1~n每个身高都放到树状数组里 
	}
	for(int i=n;i>=1;i--){
		int l=1,r=n;
		while(l<r){
			int mid=l+r>>1;
			//mid有可能被占了
			//所以不能sum(mid-1)>=w[i]
			//身高小于mid的数量 
			if(sum(mid)>=w[i]+1){
				r=mid;
			}else{
				l=mid+1;
			}
		}
		res[i]=l;
		add(l,-1);//在树状数组去除 
	}
	for(int i=1;i<=n;i++){
		cout<<res[i]<<endl;
	}
}

---

楼兰图腾 

241. 楼兰图腾 - AcWing

        先从前往后扫一遍，用树状数组统计每个点左边比它高/低的数量，然后从后往前扫一遍，树状数组统计每个点右边比它高/低的数量，相乘即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N=2e5+5;
int n;
int w[N]; 
ll up[N],down[N];
int tree[N];
int lowbit(int x){
	return x&-x;
}
void add(int x,int c){
	for(int i=x;i<=n;i+=lowbit(i)){
		tree[i]+=c;
	}
}
ll sum(int x){
	ll res=0;
	for(int i=x;i>0;i-=lowbit(i)){
		res+=tree[i];
	}
	return res;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>w[i];
	} 
	for(int i=1;i<=n;i++){
		int y=w[i];
		up[i]=sum(n)-sum(y);//左边比w[i]高的 
		down[i]=sum(y-1);//左边比w[i]低的 
		add(y,1);
	}
	memset(tree,0,sizeof tree);
	ll res1=0,res2=0;
	for(int i=n;i>=1;i--){
		int y=w[i];
		ll upr=sum(n)-sum(y);//右边比w[i]高的 
		ll downr=sum(y-1);//右边比w[i]低的 
		//以w[i]为顶点
		//V的数量=左边更高的*右边更高的
		//∧的数量=左边更低的*右边更低的
		res1+=up[i]*upr;
		res2+=down[i]*downr;
		add(y,1);
	}
	cout<<res1<<" "<<res2;
} 

---

最大上升子序列和（维护前缀区间最大值）

3662. 最大上升子序列和 - AcWing题库

        数据范围比较大，朴素方法会超时，可以用树状数组来维护前缀区间（上升序列和）的最大值

        首先将原数组W[]离散化并去重，得到位置数组，对于原数组每个数W[i]，在位置数组中找到比它小（此处为严格上升序列）的最大数的位置L，在树状数组查询得到[1~L]的最大值X，则以该数结尾的严格上升子序列的最大和为X+W[i]，并插入到树状数组L+1位置，更新[1,L+1]的最大值

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N=1e5+5;
int n,m;
int w[N];
ll tr[N];
ll res;
int lowbit(int x){
	return x&-x;
}
void modify(int x,ll c){
	for(int i=x;i<=m;i+=lowbit(i)){
		tr[i]=max(tr[i],c);
	}
}
ll query(int x){
	ll res=0;
	for(int i=x;i>0;i-=lowbit(i)){
		res=max(res,tr[i]);
	}
	return res;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++) cin>>w[i];
	int q[N];
	memcpy(q,w,sizeof w);
	sort(q+1,q+n+1);
	m=unique(q+1,q+n+1)-(q+1);
	for(int i=1;i<=n;i++){
		int l=1,r=m;
		while(l<r){
			int mid=l+r+1>>1;
			if(q[mid]<=w[i]){
				l=mid;
			}else{
				r=mid-1;
			}
		}
		ll t=query(l)+w[i];
		res=max(res,t);
		modify(l+1,t);
	}
	cout<<res;
} 

---

小朋友排队

1215. 小朋友排队 - AcWing

        对于一个小朋友，被换位置的次数 等于 前面比他高的人数 和 右面比他低的人数 之和，用树状数组来维护。最后被换sum次，则该小朋友权值为1+2+...+sum=sum*(sum+1)/2

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N=1e6+5;
int n;
int w[N];
ll tr1[N],tr2[N];
ll res;
int left_[N],right_[N];
int lowbit(int x){
	return x&-x;
}
void add(ll tr[N],int x,int c){
	for(int i=x;i<N;i+=lowbit(i)){
		tr[i]+=c;
	}
}
ll query(ll tr[N],int x){
	ll res=0;
	for(int i=x;i>0;i-=lowbit(i)){
		res+=tr[i];
	}
	return res;
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>w[i];
		w[i]++;
	}
	//前面比他高的 
	for(int i=1;i<=n;i++){
		left_[i]=query(tr1,N-1)-query(tr1,w[i]);
		add(tr1,w[i],1);
	}
	//后面比他低的 
	for(int i=n;i>=1;i--){
		right_[i]=query(tr2,w[i]-1);
		add(tr2,w[i],1);
	}
	//总共被换了 left+right次
	//1+2+...+sum =  sum*(sum+1)/2
	for(int i=1;i<=n;i++){
		int sum=left_[i]+right_[i];
		res+=(ll)sum*(sum+1)/2;
	}
	cout<<res;
} 

---

维护数组

4878. 维护数组 - AcWing题库

        开两个树状数组，分别维护min(d[i],b)和min(d[i],a) 

① 以min(d[i],b)的操作为例，当对d[i]+y时，判断此时d[i]是否已经>=b，如果>=b，则说明min(d[i],b)此后一直是b，不用操作；如果<b，则在树状数组中，将d[i]增加至min(d[i]+y,b).

② 由于树状数组维护的就是min(d[i],b)和min(d[i],a)，直接按题目要求求和即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N=2e5+5;
int k,a,b,n,q;
int tr1[N],tr2[N];
int d[N];
int lowbit(int x){
	return x&-x;
}
void add(int tr[N],int x,int c){
	for(int i=x;i<=n;i+=lowbit(i)){
		tr[i]+=c;
	}
}
ll query(int tr[N],int x){
	ll res=0;
	for(int i=x;i>0;i-=lowbit(i)){
		res+=tr[i];
	}
	return res;
}
int main(){
	cin>>n>>k>>a>>b>>q;
	while(q--){
		int c;cin>>c;
		if(c==1){
			int x,y;cin>>x>>y;
			if(d[x]<b){
				if(d[x]+y>=b) add(tr1,x,b-d[x]);
				else add(tr1,x,y);
			}
			if(d[x]<a){
				if(d[x]+y>=a) add(tr2,x,a-d[x]);
				else add(tr2,x,y);
			}
			d[x]+=y;
		}else{
			int p;cin>>p;
			ll res=query(tr1,p-1);
			res+=query(tr2,n)-query(tr2,p+k-1);
			cout<<res<<endl;
		}	
	}
}