Codeforces Global Round 22 D. Permutation Addicts

 input:

3
4
5 3 1 2
6
7 7 7 3 3 3
6
4 4 4 0 0 0

output:

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

题目大意：

给定长度为n的数组a和整数k，0<=k<=n。根据以下规则计算数组b
对于1<=i<=n，令x=a[i]
如果x<=k，令b[x]为最后一个元素a[j] (1<=j<i) ，使得a[j]>k，如果不存在该a[j]，则令b[x]=n+1；
如果x>k，令b[x]为最后一个元素a[j] (1<=j<i)，使得a[j]<=k，如果不存在该a[j]，则令b[x]=0。

现在已知b数组，求k和a数组。如果有多个a满足题意，输出任意一个。

解题思路：

首先我们先来求k，由给定的定义可知对于一个i<k,b[i]>i，因为i<=k,b[i]>k，又因为是一个全排列，所以满足这种条件的数字一定只有k个，所以我们只要统计出满足该条件的数的数量就可以求出k的值。

然后我们再来看一下这个序列，通过分析序列b的定义可知，b中一定存在一个数值为0/n+1的数字，那么这些数字对应位置的下标，也就是序列a中的元素，一定是处于a序列中最前面的那一段，而且这一段（除了最后一段外）中最后一个数字也一定会在b序列中出现，所以我们先根据b序列中值为0/n+1的数的下标找到序列a的第一段，然后在第一段中找在b序列中出现的元素，那么xuxu列的最后一个元素，而其他的元素则是可以随意排列，然后再去找下一段，直到找到最后一段，即该段中所有的数字均不会出现在b序列中，最后输出即可。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
const int N=1e5+10;
int b[N],a[N],k,n;
vector<int>p[N],ans;
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n;
		ans.clear();
		for(int i=0;i<=n+1;i++)
		p[i].clear();
		k=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			k+=(a[i]>i);
			p[a[i]].push_back(i);
		}
		int now;
		if(p[0].size())
		now=0;
		else
		now=n+1;
		while(p[now].size())
		{
			int last=now;
			int temp=now;
			for(int i=0;i<p[temp].size();i++)
			if(p[p[temp][i]].size())
			now=p[temp][i];
			else
			ans.push_back(p[temp][i]);
			if(last==now)
			break;
			ans.push_back(now);
		}
		cout<<k<<endl;
		for(int i=0;i<ans.size();i++)
		cout<<ans[i]<<" ";
		cout<<endl;
	}
	return 0;
}