输入
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;
}