题意:给定一个序列q,问p*p=q(即p[p[i]] = q[i])的p序列是否存在,不存在输出-1,反之输出任意一个p序列。
思路:置换群。
一个有奇数个的元素的群经过一次映射后,是不会改变置换后元素的顺序的(这里我们把所有元素串起来看待)。
一个有偶数个的元素的群经过一次映射后,会把群均分。
(1 2 3 4 5 6 7)^2 = (1 3 5 |7|(last) 2 4 6)。
(1 2 3 4 5 6 7 8)^2 = (1 3 5 7)(2 4 6 8)。
这样每次找到一个置换群,若元素个数have为奇,那么直接找该群的解,可以推出原序列第一个元素是群里面中间的元素,因为映射一次后顺序不变,我们依次推就可以了。若元素个数have为偶,那么若存在另一个个数为have的群,将两个群合并起来找解,从第一个位置找,每找一次跟着向后变一次。
AC代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <string>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define MAXN (1000000+10)
#define MAXM (200000+10)
#define Ri(a) scanf("%d", &a)
#define Rl(a) scanf("%lld", &a)
#define Rf(a) scanf("%lf", &a)
#define Rs(a) scanf("%s", a)
#define Pi(a) printf("%d\n", (a))
#define Pf(a) printf("%.2lf\n", (a))
#define Pl(a) printf("%lld\n", (a))
#define Ps(a) printf("%s\n", (a))
#define W(a) while(a--)
#define CLR(a, b) memset(a, (b), sizeof(a))
#define MOD 1000000007
#define LL long long
#define lson o<<1, l, mid
#define rson o<<1|1, mid+1, r
#define ll o<<1
#define rr o<<1|1
#define PI acos(-1.0)
using namespace std;
struct Node{
int val, id;
};
Node num[MAXN];
int ans[MAXN];
vector<Node> G[MAXN];
bool vis[MAXN];
int use[MAXN];
int main()
{
int n; Ri(n);
for(int i = 1; i <= n; i++)
Ri(num[i].val), num[i].id = i;
CLR(vis, false); int top = 0; CLR(use, -1); CLR(ans, -1);
for(int i = 1; i <= n; i++)
{
if(vis[i]) continue; vis[i] = true;
int next = num[i].val, now = i; int have = 1;
G[top].clear(); G[top].push_back(num[i]);
while(now != next)
{
vis[next] = true; G[top].push_back(num[next]);
next = num[next].val; have++;
}
//Pi(have);
if(have & 1)
{
int first = G[top][have/2].val;
for(int j = 0; j < have; j++)
ans[G[top][j].id] = first, first = num[first].val;
}
else
{
if(use[have] != -1)
{
int first = G[use[have]][0].id;
//Pi(last); Pi(G[top][0].id);
for(int j = 0; j < have; j++)
ans[G[top][j].id] = first, ans[first] = G[top][j].val, first = num[first].val;
use[have] = -1;
}
else
use[have] = top;
}
top++;
}
bool flag = true;
for(int i = 1; i <= n; i++) if(ans[i] == -1) {flag = false; break;}
if(!flag) Pi(-1);
else
{
printf("%d", ans[1]);
for(int i = 2; i <= n; i++)
printf(" %d", ans[i]);
printf("\n");
}
return 0;
}