Educational Codeforces Round 131 (Rated for Div. 2) D. Permutation Restoration（贪心）

题目链接：Problem - D - Codeforces

 样例输入：

4
4
0 2 0 1
2
1 1
5
0 0 1 4 1
3
0 1 3

样例输出：

2 1 4 3 
1 2 
3 4 2 1 5 
3 2 1 

题意：多组样例，每组样例给你一个n，代表数组b[]的长度，我们需要构造一组1~n的全排列a[]使得对于任意的i都有i/a[i]=b[i]成立，输出任意一组a[]

分析：我们来分析第i个位置的a[i]的可能情况，模拟几个简单的数即可发现有：

i/(b[i-1]+1)+1<=a[i]<=i/b[i],这是比较显然的，也就是说我们能够O(1)求出来每个数的可能取值的区间。

现在题目就转化为：

我们有n个可能取值的区间，其中区间左右边界均小于等于n，让我们在每个区间中选取一个数，且要保证任意两个区间内选取的数不能相同，任意构造出来一组解即可。

这是一个比较经典的问题，按照贪心的思想来想，我们首先要按照n个区间的右区间从小到大进行排序，右区间越小我们越应该先给该区间分配一个数，因为我们是从小到大考虑每个数，在右区间相同的前提下我们再按照左区间从小到大排序，这也是容易理解的，我们优先考虑 l[i] 是否能作为第i个区间所选取的数，因为在第i个区间之后的区间不一定包含l[i]，而且即使包含l[i]，那么这个区间的区间右边界也一定大于等于当前区间右边界，所以比当前区间的选数可能性更多，这是由于排序规则决定的，所以我们只需要从每个区间左边界开始遍历，一直到该区间右边界，当一个数尚未被选时我们就选他。这种套路希望大家还是掌握一下，我觉得可能会出现在各种题目中。

刚才分析了大概思路，但是这样做会TLE，我们应该稍微对它进行优化，就是记录一个pos，标记1~pos的数已经被选过，那么当我们遍历到第i个区间时我们可以从max(l[i],pos)位置开始选数。这样就能AC了

下面是代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=5e5+10;
int n,ans[N];
bool vis[N];
struct node{
	int id,l,r;
}p[N];
bool cmp(node a,node b)
{
	if(a.r!=b.r)
		return a.r<b.r;
	return a.l<b.l;
}
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			int t;
			scanf("%d",&t);
			p[i].id=i;
			p[i].l=i/(t+1)+1;
			if(t!=0)	p[i].r=i/t;
			else    p[i].r=n;
		}
		sort(p+1,p+n+1,cmp);
		memset(vis,false,sizeof vis);
		int pos=1;
		for(int i=1;i<=n;i++)
		{
			while(vis[pos]) pos++;//优化，标记1~pos的数都已经被选过 
			for(int j=max(p[i].l,pos);j<=p[i].r;j++)
			if(!vis[j])
			{
				vis[j]=true;
				ans[p[i].id]=j;
				break;
			}
		}
		for(int i=1;i<=n;i++)
			printf("%d ",ans[i]);
		puts("");
	}
	return 0;
}