CodeForces 765 D.Artsem and Saunders（构造）

Description

记$[n]$为集合$\{1,2,...,n\}$，已知映射$f:[n]\rightarrow [n]$，问是否存在映射$g:[n]\rightarrow [m],h:[m]\rightarrow [n]$满足$g(h(x))=x,\forall x\in [m],h(g(x))=f(x),\forall x\in [n]$

Input

第一行一整数$n$，之后输入$n$个整数表示$f(1),...,f(n)(1\le n\le 10^5,1\le f(i)\le n)$

Output

如果存在满足条件的正整数$m$以及映射$g,h$则输出，否则输出$-1$

Sample Input

3
1 2 3

Sample Output

3
1 2 3
1 2 3

Solution

由$g(h(x))=x$及$h(g(x))=f(x)$知，$h(x)=h(g(h(x)))=f(h(x)),\forall x\in [m]$，故$h$的值域为映射$f$的不动点集合的子集，故必有$m\le n$，如果$f$没有不动点则无解，否则设$f$的不动点集合为$\{x_1,...,x_M\}$，不妨取$h(i)=x_i,1\le i\le m$，若$m<M$，由于$h(g(x_{m+1}))=x_{m+1}\not\in \{x_1,...,x_m\}$，矛盾，故$m=M$

由$g(h(x))=x,\forall x\in [m]$知$g(x_i)=i,1\le i\le m$ ，对于$x\not\in \{x_1,...,x_m\}$，由于$h(g(x))=f(x)$，若$f(x)\not\in {x_1,...,x_m}$则无解，否则假设$f(x)=x_j$，则$g(x)=j$，以此求出所有的$g(x)$即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 111111
int m,f[maxn],g[maxn],h[maxn],pos[maxn];
int main()
{
    while(~scanf("%d",&m))
    {
        memset(g,-1,sizeof(g));
        memset(pos,-1,sizeof(pos));
        int n=0;
        for(int i=1;i<=m;i++)
        {
            scanf("%d",&f[i]);
            if(f[i]==i)h[++n]=i,g[i]=n,pos[i]=i;
        }
        if(!n)printf("-1\n");
        else
        {
            int gg=0;
            for(int i=1;i<=m;i++)
            {
                if(g[i]==-1)
                {
                    if(pos[f[i]]==-1)gg=1;
                    else g[i]=g[pos[f[i]]];
                }
            }
            if(gg)printf("-1\n");
            else
            {   
                printf("%d\n",n);
                for(int i=1;i<=m;i++)printf("%d%c",g[i],i==m?'\n':' ');
                for(int i=1;i<=n;i++)printf("%d%c",h[i],i==n?'\n':' ');
            }
        }
    }
    return 0;
}