链接:https://codeforces.com/contest/1530/problem/D
题意:n个人,每个人希望给第ki个人送礼物,但不能出现一个人接受多份礼物的情况,不能给自己送礼物,问礼物怎么分配能满足最多人的愿望。
思路:
比较容易想到的是贪心,扫一遍,满足第一个想给k送礼物的人,然后后面遇到愿望相同的就跳过,最后把剩下的随意分配给还没分配那些人。
但是这样做有可能出现自己给自己送礼物的情况,所以关键的步骤就是去重。
去重方法:扫一遍,如果出现了自己给自己送礼物的情况,则满足那个送自己礼物的人的愿望,同时令原本匹配上的人给送自己礼物的人送礼物(由于送自己礼物的人愿望必不可能是自己,而原本匹配的人与他并不是同一个人,所以不会产生送自己礼物的情况),由于送给自己礼物的那个人满足了自己的愿望,所以匹配数不会减少。
#include<bits/stdc++.h>
using namespace std;
const int N=200005;
int n,b[N],a[N],vis[N],s,y[N];
void sol(){
scanf("%d",&n);
for(int i=1;i<=n;i++)
a[i]=b[i]=vis[i]=y[i]=0;
s=0;
for(int i=1;i<=n;i++){
scanf("%d",&b[i]);
if(!vis[b[i]]){
a[i]=b[i];
vis[b[i]]=1;
y[b[i]]=i;
s++;
}
}
int p=1;
for(int i=1;i<=n;i++){
if(!a[i]){
while(vis[p])
p++;
vis[p]=1;
a[i]=p;
y[p]=i;
}
}
for(int i=1;i<=n;i++){ //去重
if(a[i]==i){
int t=y[b[i]];
swap(a[i],a[t]);
swap(y[b[i]],y[i]);
}
}
printf("%d\n",s);
for(int i=1;i<=n;i++)
printf("%d ",a[i]);
puts("");
}
int main(){
int t;
scanf("%d",&t);
while(t--)
sol();
return 0;
}