来自学长的快乐AK题——Day2 C

C——维护第k大的数

Description

给出一个长度为n的序列A描述一个长度为n的全排列a，A_i表示前i-1个数中有多少个数比a_i大。
请求出这个全排列a

思路

从后往前遍历A，根据题目中的定义，a的前n-1个数中有A_n个数比a_n大，换言之，a_n就是1~n中第A_n+1大的数。同理，a_n-1就是除去已经确定的a_n后第A_n-1+1大的数。以此类推，就将问题转化为如何维护一个序列中的第k大数。

算法还是有很多，例如快排（nlogn），归并(nlogn或n)，我是采用更好写也更快的桶排(n)，卡到了O(n²)。但数据一大（超时）或一恶心（超出桶的范围），桶排就不顶用了。

学长教了一个更优秀的算法：利用树状数组动态维护前缀和。

对于每个存在的数i，将数组中下标为i的位置赋值为1。对于任意一个数i，只要求出数组下标为1~i的前缀和，就能得到i的大小排名。如果这个数不存在或被删除，将其对应下标的位置赋值为0即可。
原理也容易明白，就是优化了桶排，通过前缀和快速求出有多少个数小于i。

有了这个东东，我们只需要二分数组的下标，得到前缀和正好为k时的下标，即为第k大的数。

代码

（我懒得再写树状数组的优化了，反正我都过了）

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1e4;
int n,a[N],vis[N],ans[N];
int query(int k)
{
	for(int i=n,cnt=0;i;i--)
	{
		if(!vis[i]) cnt++;
		if(cnt==k) return i;
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	for(int i=n;i;i--)
	ans[i]=query(a[i]+1),vis[ans[i]]=1;
	for(int i=1;i<=n;i++)
	printf("%d ",ans[i]);
	return 0;
}