CodeForces 765D Artsem and Saunders

题目链接：http://codeforces.com/contest/765/problem/D
题意：有这么一个函数f:n->n，问你是否存在g：n->m，和h:m->n,且满足g(h(x)) = x,h(g(x)) = f(x),现告诉你n和f(1)到f(n),问你是否存在，若不存在则输出-1，若存在输出m，g(1)到g(n)和h(1)到h(m)
解析：
因为g(h(x)) = x,h(g(x)) = f(x)，那么我们现在假设有一个y = h(x)，
所以对于h(g(y)) = f(y)就有h(g(h(x))) = f(h(x))，
所以就有f(h(x)) = h(x)
所以就有f(x) = x
这样的一段我觉得正确的推理下来的话，如果存在g和f那么必定有x=f(x)，于是这样子h(x)就构造出来了，那么通过g(h(x)) = x，也把g(x)构造出来了，剩下的就是判断合法性了，也就是g(x) = g(f(x))

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+100;
int h[maxn];
int g[maxn];
int f[maxn];
int main()
{
    int n;
    scanf("%d",&n);
    int k = 1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&f[i]);
        if(i == f[i])           //当x==f(x)的时候开始构造h(x)
        {
            h[k] = i;
            g[h[k]] = k;
            k++;
        }
    }
    int flag = 0;
    for(int i=1;i<=n;i++)
    {
        if(!g[f[i]])        //如果能构造出来，那么g(f(x))一定存在
        {
            flag = 1;
            break;
        }
    }
    if(flag)
        puts("-1");
    else
    {
        printf("%d\n",k-1);
        for(int i=1;i<=n;i++)
            printf("%d ",g[f[i]]);
        puts("");
        for(int i=1;i<k;i++)
            printf("%d ",h[i]);
        puts("");
    }
    return 0;
}