[백준] 9322번 철벽 보안 알고리즘

💡문제접근

• 제 1 공개키
• 제 2 공개키는 제 1 공개키의 단어 재배치

💡테스트케이스

입력
A B C D
D A B C
C B A P

출력
B A P C

Key : 제 2 공개키, Value : 암호문
사전 : {"D" : "C", "A" : "B", "B" : "A", "C" : "P"}

A → B, B → A, C → P, D → C

💡코드(메모리 : 34112KB, 시간 : 124ms)

from collections import defaultdict
import sys
input = sys.stdin.readline

T = int(input())
for _ in range(T):
    n = int(input())
    a = list(input().split())
    b = list(input().split())
    c = list(input().split())
    encryption_dict = defaultdict(int)
    for i in range(n):
        encryption_dict[b[i]] = c[i]
    for i in a:
        print(encryption_dict[i], end = " ")

💡소요시간 : 24m