[BOJ 22858] 원상 복구 (small)

https://www.acmicpc.net/problem/22858

22858번: 원상 복구 (small)

규칙에 따라 섞인 수열의 맨 처음 형태를 구하면 된다.

 

$P_i$에서 $i=D_i$인 경우 $S_i$가 된다.

즉, $P[D[i]]=S[i]$이다. 이를 $K$번 반복해주면 된다. $NK\leq 10^7$이기에 전부 돌려봐도 해결 가능하다.

int N, K, S[MAXN], D[MAXN], P[MAXN];
int main(void) {
    fastio;
    cin >> N >> K;
    for (int i = 1; i <= N; i++) cin >> S[i];
    for (int i = 1; i <= N; i++) cin >> D[i];
    while (K--) {
        for (int i = 1; i <= N; i++) P[D[i]] = S[i];
        for (int i = 1; i <= N; i++) S[i] = P[i];
    }
    for (int i = 1; i <= N; i++) cout << P[i] << ' ';
    return 0;
}