树状数组&逆序对

任意给定一个集合a,用t[val]保存数值val1在集合a内出现的次数，那么数组t在[l,r]上的区间和${\sum }_{i=1}^{r}t[i]$就表示集合a中范围在[l,r]内的数有多少个。可以在集合a的数值范围上建立一个树状数组来维护t的前缀和。

对于给定的序列a，倒序扫描之，对于每个数a[i]:

1. 在树状数组中查询前缀和[1,a[i]-1]，累加到答案ans中（这些数比a[i]小，位置却在它之后）。
2. 单点修改，t[a[i]]++，同时维护前缀和。

数值范围较大时可以先离散化。

---

241. 楼兰图腾

第一遍输入同时正序扫描，可以保存下每个数前面比它大的数的个数和比它小的数的个数。
第二遍逆序扫描，求出每个数后面比它大的数的个数和比它小的数的个数。
交叉相乘，累加得结果。

#include<bits/stdc++.h>
using namespace std;

const int N=2e5+10;
int a[N],c[2][N],n;
long long ans1=0,ans2=0;
long long mem1[N],mem2[N];

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
	return x*f;
}

int lowbit(int x)
{
	return x&(-x);
}

void add(int k,int pos)
{
	for(int i=pos;i<=n;i+=lowbit(i))
		c[k][i]++;
}

int get(int k,int pos)
{
	int sum=0;
	for(int i=pos;i;i-=lowbit(i))
		sum+=c[k][i];
	return sum;
}

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		mem1[i]=get(0,a[i]-1);
		mem2[i]=get(0,n)-get(0,a[i]);
		add(0,a[i]);
	}
	for(int i=n;i>=1;i--)
	{
		long long x=get(1,a[i]-1);
		long long y=get(1,n)-get(1,a[i]);
		ans1+=y*mem2[i];
		ans2+=x*mem1[i];
		add(1,a[i]);
	}
	cout<<ans1<<' '<<ans2<<endl;
    return 0;
}

---

244. 谜一样的牛

如果第k头牛前面有$A_k$头比它矮，那么它的身高$H_k$是数值1~n中第$A_k+1$小的没有在$H_{k+1},H_{k+2},...H_n$中出现过的数。

需要实时维护一个01序列，支持查询第k个1的位置，以及修改序列中的任一数值。

用树状数组c维护01序列的前缀和，每次查询时二分答案，通过ask(mid)可以得到前mid个数中有多少个1，与k比较大小，可以确定二分上下界的范围。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int a[N],c[N],ans[N],n;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
	return x*f;
}

int lowbit(int x)
{
	return x&(-x);
}

void add(int pos,int x)
{
	for(int i=pos;i<=n;i+=lowbit(i))
		c[i]+=x;
}

int get(int pos)
{
	int sum=0;
	for(int i=pos;i;i-=lowbit(i))
		sum+=c[i];
	return sum;
}

int main()
{
	cin>>n;
	add(1,1);
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&a[i]);
		add(i,1);
	}
	for(int i=n;i>=1;i--)
	{
		int l=1,r=n;
		while(l<r)
		{
			int mid=(l+r)>>1;
			if(get(mid)<a[i]+1) l=mid+1;
			else r=mid;
		}
		ans[i]=r;
		add(r,-1);
	}
	for(int i=1;i<=n;i++)
        cout<<ans[i]<<endl;
    return 0;
}