2021-08-10

题目描述

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

输入

第一行一个正整数n
第二行给出序列A

输出

对应的全排列

输入样例

3
0 0 1

输出样例

1 3 2

说明

30%:
2<=n<=20
100%:
2<=N<=8000

题解：

从后往前走，每个位置上的数为第 a_i + 1 大的还存在的数

O ( n² )

Code

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
#define LL long long
#define ull unsigned LL
#define db double
#define ldb long db
#define re register
using namespace std;

const int N=1e6+100,M=2020,mod=1000000007;
int n,a[N],f[N],fg[N];
int find ( int x )
{
	for ( int i=n,j=0;i>=1;i-- )
	{
		if ( !fg[i] ) j++;
		if ( j==x ) { fg[i]=1;return i; }
	}
	return -1;
}

int main ( )
{
	scanf ( "%d",&n );
	for ( int i=1;i<=n;i++ ) scanf ( "%d",&a[i] );
	for ( int i=n;i>=1;i-- ) f[i]=find ( a[i]+1 );
	for ( int i=1;i<=n;i++ ) printf ( "%d ",f[i] ); 
	return 0;
}

而当数据升级到 1e5 时，就不行了

则又有个优化，用树状数组优化查询

复杂度为 O ( n log₂²n )