2022牛客多校 7 C Constructive Problems Never Die

本文链接：https://blog.csdn.net/weixin_54562114/article/details/126235533

输入

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

输出

NO
YES
1 3 2
YES
4 5 1 2 3 6

题目大意：给我们一个长度为n的序列a，让我们找到一个1~n的全排列p使得，任意的p[i]!=a[i]，如果不存在输出NO，存在输出YES，然后输出一个满足条件的序列。

解题思路：首先我们可以设一个可供选数的区间1~n,然后在选前n-1个数时有两种选择，即可以选l，也可以选r，两种选择选一个即可，然后问题就出在了最后一个数的选择上，因为此时我们只剩下一个数可以选了，所以是有可能会与原序列中的数字相同的，所以我们需要特判一下，如果相同，我们就可以考虑将这个数与前面某个位置满足a[i]!=p[n] 的数p[i]交换，为什么这是可以的呢？首先因为是一个全排列，所以p[i]!=p[n]，所以a[n]!=p[i]，又因为我们在寻找的时候使得a[i]!=p[n],所以交换之后恰好可以满足条件。

上代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <map>
using namespace std;
const int N=1E5+10;
int n,a[N],ans[N],b[N];
map<int,int>mp;
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n;
		mp.clear();
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
			mp[a[i]]++;
		}
		if(mp[a[1]]==n)
		{
			cout<<"NO"<<endl;
			continue;
		}
		int l=1,r=n;
		for(int i=1;i<n;i++)
		{
			if(a[i]!=l)
			{
				ans[i]=l;
				l++;
			}
			else
			{
				ans[i]=r;
				r--;
			}
		}
		ans[n]=l;
		if(ans[n]==a[n])
		{
			for(int i=1;i<=n-1;i++)
			if(ans[n]!=a[i])
			swap(ans[n],ans[i]);
		}
			cout<<"YES"<<endl;
			for(int i=1;i<=n;i++)
			cout<<ans[i]<<" ";
			cout<<endl;

	}
	return 0;
}