[백준,c++] 22858번 - 원상 복구(small)

문제

22858번: 원상 복구 (small)

 

코드

#include <iostream>
#include <memory.h>

using namespace std;

int D[10001], S[10001], v[10001];
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    
    int N, K; 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++){
            v[D[i]] = S[i];
        }
        memcpy(S,v,sizeof(S));
    }
    for(int i=1; i<=N; i++) cout<< v[i] << ' ';
}

 

풀이

문제를 잘 이해해야 합니다.

위의 P[1 4 5 3 2] 가 적힌 카드가 D 순서대로 셔플을 진행했을 때 S[3 5 1 4 2]가 나오게 되는 문제인데 조금 더 풀어서 이야기하면 다음과 같습니다.

• P = 1 일때 D = 4입니다. 따라서 S배열의 4번째 인덱스에 넣으면 됩니다.
• P = 4일때 D = 3입니다. 따라서 S배열의 3번째 인덱스에 넣으면 됩니다.
• P = 5일때 D = 1입니다. 따라서 S배열의 1번째 인덱스에 넣으면 됩니다.
• P = 3일때 D = 2입니다. 따라서 S배열의 2번째 인덱스에 넣으면 됩니다.
• P = 2일때 D = 5입니다. 따라서 S배열의 5번째 인덱스에 넣으면 됩니다.

따라서 위 로직을 기준으로 단순 구현을 해주면 됩니다.