树状数组

题目分享之前，先学习一下树状数组，阅读文章

---

Luogu：

P3374 【模板】树状数组 1

P3368 【模板】树状数组 2

以上两题相关内容详见 这篇文章

---

Acwing：

241.楼兰图腾

题面

算法分析

根据题中描述的"v"和"^"的概念，我们可以相应地统计出任意一个位置$i$左边和右边满足要求的个数。下面以"v"为例：

对于"v"形，要求对于一个位置$i$的值$y_i$左右两边都要大于它，所以计算出相应的 $left\_greater[i]$ 和 $right\_greater[i]$ ，最终的结果 $$ans1=\sum _{i=1}^{n}left\_greater[i]\ast right\_greater[i]$$

对于"^"形，一样的道理$$ans2=\sum _{i=1}^{n}left\_lower[i]\ast right\_lower[i]$$

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define rg register
using namespace std;
typedef long long ll;
inline ll sread()
{
	ll x=0,f=1;char c=getchar();
	while(c>'9'||c<'0') {if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
	return f*x; 
}
const ll maxn=5e5+10;
ll n,c[maxn],a[maxn],ans1,ans2;
ll left_lower[maxn],left_greater[maxn];
ll right_lower[maxn],right_greater[maxn];
ll lowbit(ll x)
{
	return x&(-x);
}
void add(ll x,ll val)
{
	for(rg ll i=x;i<=n;i+=lowbit(i)) c[i]+=val;
}
ll query(ll x)
{
	ll sum=0;
	for(rg ll i=x;i;i-=lowbit(i)) sum+=c[i];
	return sum;
}
int main()
{
	n=sread();
	for(rg ll i=1;i<=n;i++) a[i]=sread();
	
	//从左往右 
	for(rg ll i=1;i<=n;i++)
	{
		left_lower[i] = query(a[i]-1);
		left_greater[i] = query(n) - query(a[i]);
		add(a[i],1);
	}
	
	memset(c,0,sizeof(c));
	 
	//从右往左 
	for(rg ll i=n;i>=1;i--)
	{
		right_lower[i] = query(a[i]-1);
		right_greater[i] = query(n) - query(a[i]);
		add(a[i],1);
	}
	
	for(rg ll i=1;i<=n;++i)
	{
		ans1 += left_greater[i] * right_greater[i];
		ans2 += left_lower[i] * right_lower[i];
	} 
	printf("%lld %lld\n",ans1,ans2);
	return 0;
}

---

242.一个简单的整数问题

题面

算法分析

区间修改，单点查询。模板题，不再多说

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define rg register
using namespace std;
typedef long long ll;
inline int sread()
{
	int x=0,f=1;char c=getchar();
	while(c>'9'||c<'0') {if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
	return f*x; 
}
const int maxn=5e5+10;
int a[maxn],x,y,k,n,m,ans,c[maxn];
char ops;
int lowbit(int x)
{
	return x&(-x);
}
void update(int x,int val)
{
	for(rg int i=x;i<=n;i+=lowbit(i)) c[i]+=val;
}
int getsum(int x)
{
	int sum=0;
	for(rg int i=x;i;i-=lowbit(i)) sum+=c[i];
	return sum;
}
int main()
{
	n=sread();	m=sread();
	for(rg int i=1;i<=n;++i)
	{
		a[i]=sread();
	}
	for(rg int i=1;i<=m;++i)
	{
		cin>>ops;
		if(ops=='C') 
		{
			x=sread();	y=sread();	k=sread();
			update(x,k); update(y+1,(-1)*k);
		}
		else if(ops=='Q')
		{
			x=sread();
			ans=a[x]+getsum(x);
			printf("%d\n",ans);
		}
	} 
	return 0;
}

---

243.一个简单的整数问题2

题面

算法分析

区间修改，区间查询。模板题，不再多说。

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define rg register
using namespace std;
typedef long long ll;
inline ll sread()
{
	ll x=0,f=1;char c=getchar();
	while(c>'9'||c<'0') {if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
	return f*x; 
}
const ll maxn=5e5+10;
ll a[maxn],x,y,k,n,m,ans,c[3][maxn],sum[maxn];
char ops;
ll lowbit(ll x)
{
	return x&(-x);
}
void update(ll k,ll x,ll val)
{
	for(rg ll i=x;i<=n;i+=lowbit(i)) c[k][i]+=val;
}
ll getsum(ll k,ll x)
{
	ll sum=0;
	for(rg ll i=x;i;i-=lowbit(i)) sum+=c[k][i];
	return sum;
}
int main()
{
	n=sread();	m=sread();
	for(rg ll i=1;i<=n;++i)
	{
		a[i]=sread();
		sum[i]=sum[i-1]+a[i];
	}
	for(rg ll i=1;i<=m;++i)
	{
		cin>>ops;
		if(ops=='C') 
		{
			x=sread();	y=sread();	k=sread();
			update(1,x,k); update(1,y+1,(-1)*k);
			update(2,x,x*k);	update(2,y+1,(-1)*(y+1)*k);
		}
		else if(ops=='Q')
		{
			x=sread(); y=sread();
			ans=sum[y]+(y+1)*getsum(1,y)-getsum(2,y);
			ans-=sum[x-1]+x*getsum(1,x-1)-getsum(2,x-1); 
			printf("%lld\n",ans);
		}
	} 
	return 0;
}

---

244.谜一样的牛

题面

算法分析

我们发现，如果说第$K$头牛的前面有$A_k$头牛比它矮，那么它的身高$H_k$就是数值$1-n$中第$A_k+1$小的没有在$H_{k+1},H_{k+2},\dots ,H_n$中出现过的数。
所以说，我们需要建立一个长度为$n$的序列，刚开始都是$1$，然后$n$到$1$倒序扫描每一个$A_i$，对于每个$A_i$执行查询和修改操作，最后正序输出答案。

Code

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define rg register
#define lowbit(x) x&(-x)
using namespace std;
typedef long long ll;
inline int sread()
{
	int x=0,f=1;char c=getchar();
	while(c>'9'||c<'0') {if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
	return f*x; 
}
const int maxn=5e5+10;
int a[maxn],n,m,ans[maxn],c[maxn];
void update(int x,int val)
{
	for(rg int i=x;i<=n;i+=lowbit(i)) c[i]+=val;
}
int query(int x)
{
	int sum=0;
	for(rg int i=x;i;i-=lowbit(i)) sum+=c[i];
	return sum;
}
int main()
{
	n=sread();
	update(1,1);
	for(rg int i=2;i<=n;i++)
	{
		a[i]=sread();
		update(i,1);
	}
	for(rg int i=n;i>=1;i--)
	{
		int l=1,r=n;
		while(l<r)
		{
			int mid=(l+r)>>1;
			if(query(mid)<a[i]+1)
				l=mid+1;
			else r=mid;
		}
		ans[i]=r;
		update(r,-1);
	}
	for(rg int i=1;i<=n;i++)
	{
		printf("%d\n",ans[i]);
	}
	return 0;
}

---

反思与总结

1、树状数组主要可用来解决单点/区间修改和查询问题。在这方面和线段树有相似之处，但树状数组码量较小，较易理解。

2、学习了 lowbit函数的用法，体会到了二进制在树状数组实现中的关键地位。

3、在完成习题的过程中学习到了一些东西，同时发现了树状数组也有时会和其他算法结合在一起使用。