A,B就不写了。
C:
每个人都要送别人一个礼物,不能送给自己。一个人必须收到一个礼物且送出一个礼物。已知有一些人明确要送给谁礼物了,但有些人不明确。给出任意一种可行的方案。保证一定有解。
#include <bits/stdc++.h>
using namespace std;
const int N=2e5+5;
int n,tot,tot2,cnt;
int a[N],b[N],pos[N],c[N],id[N],ans[N];
bool ff[N],vis[N];
int main(){
scanf("%d",&n);
for (register int i=1; i<=n; ++i) scanf("%d",&a[i]);
for (register int i=1; i<=n; ++i) if (!a[i]) b[++tot]=i;
for (register int i=1; i<=n; ++i) if (a[i]) pos[a[i]]=i;
for (register int i=1; i<=n; ++i) if (!pos[i]) c[++tot2]=i;
for (register int i=1; i<=n; ++i)
if (!a[i] && !pos[i])
{
cnt++;
id[cnt]=i;
}
for (register int i=1; i<=n; ++i) ans[i]=a[i];
if (cnt>1)
{
for (register int i=2; i<=cnt; ++i) ans[id[i]]=id[i-1];
ans[id[1]]=id[cnt];
}
else if (cnt==1)
{
for (register int i=1; i<=n; ++i)
if (!a[i] && id[1]!=i) {ans[i]=id[1]; break;}
}
for (register int i=1; i<=n; ++i) vis[ans[i]]=true;
tot=0;
for (register int i=1; i<=n; ++i) if (!vis[i]) b[++tot]=i;
int j=1;
for (register int i=1; i<=n; ++i)
{
if (!ans[i]) ans[i]=b[j],j++;
}
for (register int i=1; i<=n; ++i) printf("%d ",ans[i]);
return 0;
}