코딩하는 덕구 🐶

122. [삼성기출18]Python 백준 5373 번 큐빙 자세한 설명 본문

알고리즘 문제 풀이

122. [삼성기출18]Python 백준 5373 번 큐빙 자세한 설명

코딩하는 덕구 🐶 2023. 5. 31. 14:29
728x90

안녕하세요 코딩하는 덕구입니다.

백준 5373 번 큐빙 입니다. 이번 문제는 코드 자체의 난이도 보다 접근 방식이 매우 중요합니다.

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

 

5373번: 큐빙

각 테스트 케이스에 대해서 큐브를 모두 돌린 후의 윗 면의 색상을 출력한다. 첫 번째 줄에는 뒷 면과 접하는 칸의 색을 출력하고, 두 번째, 세 번째 줄은 순서대로 출력하면 된다. 흰색은 w, 노란

www.acmicpc.net

먼저 백준 5373 번 큐빙 정답 코드입니다. 설명은 아래에 있습니다.

from copy import deepcopy
U = []
D = []
F = []
B = []
L = []
R = []

def init():
    global U, D, F, B, L, R
    U = [['w' for _ in range(3)] for _ in range(3)]
    D = [['y' for _ in range(3)] for _ in range(3)]
    F = [['r' for _ in range(3)] for _ in range(3)]
    B = [['o' for _ in range(3)] for _ in range(3)]
    L = [['g' for _ in range(3)] for _ in range(3)]
    R = [['b' for _ in range(3)] for _ in range(3)]

def rotate(o):
    if o == 'U':
        tmp = deepcopy(U)
        U[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        U[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        U[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(F[0])
        F[0] = R[0]
        R[0] = B[0]
        B[0] = L[0]
        L[0] = tmp
        return

    elif o == 'D':
        tmp = deepcopy(D)
        D[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        D[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        D[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(F[2])
        F[2] = L[2]
        L[2] = B[2]
        B[2] = R[2]
        R[2] = tmp
        return

    elif o == 'F':
        tmp = deepcopy(F)
        F[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        F[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        F[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(U[2])
        U[2] = [L[2][2], L[1][2], L[0][2]]
        L[2][2], L[1][2], L[0][2] = D[0][2], D[0][1], D[0][0]
        D[0][2], D[0][1], D[0][0] = R[0][0], R[1][0], R[2][0]
        R[0][0], R[1][0], R[2][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'B':
        tmp = deepcopy(B)
        B[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        B[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        B[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(U[0])
        U[0] = [R[0][2], R[1][2], R[2][2]]
        R[0][2], R[1][2], R[2][2] = D[2][2], D[2][1], D[2][0]
        D[2][2], D[2][1], D[2][0] = L[2][0], L[1][0], L[0][0]
        L[2][0], L[1][0], L[0][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'L':
        tmp = deepcopy(L)
        L[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        L[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        L[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy([U[0][0], U[1][0], U[2][0]])
        U[0][0], U[1][0], U[2][0] = B[2][2], B[1][2], B[0][2]
        B[2][2], B[1][2], B[0][2] = D[0][0], D[1][0], D[2][0]
        D[0][0], D[1][0], D[2][0] = F[0][0], F[1][0], F[2][0]
        F[0][0], F[1][0], F[2][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'R':
        tmp = deepcopy(R)
        R[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        R[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        R[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy([U[0][2], U[1][2], U[2][2]])
        U[0][2], U[1][2], U[2][2] = F[0][2], F[1][2], F[2][2]
        F[0][2], F[1][2], F[2][2] = D[0][2], D[1][2], D[2][2]
        D[0][2], D[1][2], D[2][2] = B[2][0], B[1][0], B[0][0]
        B[2][0], B[1][0], B[0][0] = tmp[0], tmp[1], tmp[2]

        return

def print_cube_upside():
    for i in range(3):
        string = ''
        for j in range(3):
            string += U[i][j]
        print(string)


def print_all(): #to debug
    for i in range(3):
        string = ''
        for j in range(3):
            string += U[i][j]
        print(string)

    for i in range(3):
        f = ''
        r = ''
        b = ''
        l = ''
        for j in range(3):
            f += F[i][j]
            r += R[i][j]
            b += B[i][j]
            l += L[i][j]
        print(f, r, b, l)

    for i in range(3):
        string = ''
        for j in range(3):
            string += D[i][j]
        print(string)

#t = 0
for test in range(int(input())):
    #t += 1
    #print('test case : ', t)
    init()
    n = int(input())
    order = input().split()
    index = 0
    for o in order:
        index += 1
        cnt = 3
        if o[1] == "+":
            cnt = 1
        for _ in range(cnt):
            rotate(o[0])

        #print(index)
        #print_all()
    print_cube_upside()

 

문제 접근

1. 큐브의 동작가능한 모든 경우의 수 = 6개의 면을 왼쪽 혹은 오른쪽으로 돌리는 경우의 수 = 12

2. 어떤 면을 왼쪽으로 1번 돌리는 것은 오른쪽으로 3번 돌리는 것과 같습니다.

즉 6가지의 경우만 작성해주고 회전 횟수를 조절하는 방식으로 구현의 피로를 낮춥니다.

3. 모든 경우를 하나도 틀리지 않고 작성해야되므로 모든 큐브의 면을 확인할 수 있는 함수를 작성해서

큐브가 돌아가는 상태를 눈으로 직접 확인한다면 모든 코드를 확인하지 않고, 문제가 생긴 코드 부분을 빠르게 찾고 고칠 수 있습니다.

 

Python 코드 구현

큐브의 모든 면을 정의하고, 큐브를 초기화 하는 함수를 작성합니다.

from copy import deepcopy
U = []
D = []
F = []
B = []
L = []
R = []

def init():
    global U, D, F, B, L, R
    U = [['w' for _ in range(3)] for _ in range(3)]
    D = [['y' for _ in range(3)] for _ in range(3)]
    F = [['r' for _ in range(3)] for _ in range(3)]
    B = [['o' for _ in range(3)] for _ in range(3)]
    L = [['g' for _ in range(3)] for _ in range(3)]
    R = [['b' for _ in range(3)] for _ in range(3)]

 

큐브의 회전할 면을 입력 받고 오른쪽으로 회전시키는 함수 rotate()를 작성합니다.

def rotate(o):
    if o == 'U':
        tmp = deepcopy(U)
        U[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        U[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        U[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(F[0])
        F[0] = R[0]
        R[0] = B[0]
        B[0] = L[0]
        L[0] = tmp
        return

    elif o == 'D':
        tmp = deepcopy(D)
        D[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        D[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        D[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(F[2])
        F[2] = L[2]
        L[2] = B[2]
        B[2] = R[2]
        R[2] = tmp
        return

    elif o == 'F':
        tmp = deepcopy(F)
        F[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        F[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        F[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(U[2])
        U[2] = [L[2][2], L[1][2], L[0][2]]
        L[2][2], L[1][2], L[0][2] = D[0][2], D[0][1], D[0][0]
        D[0][2], D[0][1], D[0][0] = R[0][0], R[1][0], R[2][0]
        R[0][0], R[1][0], R[2][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'B':
        tmp = deepcopy(B)
        B[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        B[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        B[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(U[0])
        U[0] = [R[0][2], R[1][2], R[2][2]]
        R[0][2], R[1][2], R[2][2] = D[2][2], D[2][1], D[2][0]
        D[2][2], D[2][1], D[2][0] = L[2][0], L[1][0], L[0][0]
        L[2][0], L[1][0], L[0][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'L':
        tmp = deepcopy(L)
        L[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        L[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        L[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy([U[0][0], U[1][0], U[2][0]])
        U[0][0], U[1][0], U[2][0] = B[2][2], B[1][2], B[0][2]
        B[2][2], B[1][2], B[0][2] = D[0][0], D[1][0], D[2][0]
        D[0][0], D[1][0], D[2][0] = F[0][0], F[1][0], F[2][0]
        F[0][0], F[1][0], F[2][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'R':
        tmp = deepcopy(R)
        R[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        R[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        R[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy([U[0][2], U[1][2], U[2][2]])
        U[0][2], U[1][2], U[2][2] = F[0][2], F[1][2], F[2][2]
        F[0][2], F[1][2], F[2][2] = D[0][2], D[1][2], D[2][2]
        D[0][2], D[1][2], D[2][2] = B[2][0], B[1][0], B[0][0]
        B[2][0], B[1][0], B[0][0] = tmp[0], tmp[1], tmp[2]

        return

 

큐브의 윗면을 출력하는 함수를 작성합니다.

def print_cube_upside():
    for i in range(3):
        string = ''
        for j in range(3):
            string += U[i][j]
        print(string)

 

 테스트 케이스를 입력받고, 큐브를 회전시키는 함수를 작성합니다.

회전이 끝나고나면, 큐브의 윗면을 출력합니다.

이때 -표시라면 큐브를 시계방향으로 회전시키는 rotate함수를 3번 실행합니다.

#t = 0
for test in range(int(input())):
    #t += 1
    #print('test case : ', t)
    init()
    n = int(input())
    order = input().split()
    index = 0
    for o in order:
        index += 1
        cnt = 3
        if o[1] == "+":
            cnt = 1
        for _ in range(cnt):
            rotate(o[0])

        #print(index)
        #print_all()
    print_cube_upside()

 

주석처리된 부분을 지우면 print_all 함수를 이용해 큐브의 모든 면을 확인하실 수 있습니다. 

def print_all(): #to debug
    for i in range(3):
        string = ''
        for j in range(3):
            string += U[i][j]
        print(string)

    for i in range(3):
        f = ''
        r = ''
        b = ''
        l = ''
        for j in range(3):
            f += F[i][j]
            r += R[i][j]
            b += B[i][j]
            l += L[i][j]
        print(f, r, b, l)

    for i in range(3):
        string = ''
        for j in range(3):
            string += D[i][j]
        print(string)

 

파이썬 백준 5373 번 큐빙 최종 코드입니다. 

from copy import deepcopy
U = []
D = []
F = []
B = []
L = []
R = []

def init():
    global U, D, F, B, L, R
    U = [['w' for _ in range(3)] for _ in range(3)]
    D = [['y' for _ in range(3)] for _ in range(3)]
    F = [['r' for _ in range(3)] for _ in range(3)]
    B = [['o' for _ in range(3)] for _ in range(3)]
    L = [['g' for _ in range(3)] for _ in range(3)]
    R = [['b' for _ in range(3)] for _ in range(3)]

def rotate(o):
    if o == 'U':
        tmp = deepcopy(U)
        U[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        U[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        U[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(F[0])
        F[0] = R[0]
        R[0] = B[0]
        B[0] = L[0]
        L[0] = tmp
        return

    elif o == 'D':
        tmp = deepcopy(D)
        D[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        D[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        D[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(F[2])
        F[2] = L[2]
        L[2] = B[2]
        B[2] = R[2]
        R[2] = tmp
        return

    elif o == 'F':
        tmp = deepcopy(F)
        F[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        F[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        F[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(U[2])
        U[2] = [L[2][2], L[1][2], L[0][2]]
        L[2][2], L[1][2], L[0][2] = D[0][2], D[0][1], D[0][0]
        D[0][2], D[0][1], D[0][0] = R[0][0], R[1][0], R[2][0]
        R[0][0], R[1][0], R[2][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'B':
        tmp = deepcopy(B)
        B[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        B[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        B[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy(U[0])
        U[0] = [R[0][2], R[1][2], R[2][2]]
        R[0][2], R[1][2], R[2][2] = D[2][2], D[2][1], D[2][0]
        D[2][2], D[2][1], D[2][0] = L[2][0], L[1][0], L[0][0]
        L[2][0], L[1][0], L[0][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'L':
        tmp = deepcopy(L)
        L[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        L[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        L[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy([U[0][0], U[1][0], U[2][0]])
        U[0][0], U[1][0], U[2][0] = B[2][2], B[1][2], B[0][2]
        B[2][2], B[1][2], B[0][2] = D[0][0], D[1][0], D[2][0]
        D[0][0], D[1][0], D[2][0] = F[0][0], F[1][0], F[2][0]
        F[0][0], F[1][0], F[2][0] = tmp[0], tmp[1], tmp[2]
        return

    elif o == 'R':
        tmp = deepcopy(R)
        R[0] = [tmp[2][0], tmp[1][0], tmp[0][0]]
        R[1] = [tmp[2][1], tmp[1][1], tmp[0][1]]
        R[2] = [tmp[2][2], tmp[1][2], tmp[0][2]]

        tmp = deepcopy([U[0][2], U[1][2], U[2][2]])
        U[0][2], U[1][2], U[2][2] = F[0][2], F[1][2], F[2][2]
        F[0][2], F[1][2], F[2][2] = D[0][2], D[1][2], D[2][2]
        D[0][2], D[1][2], D[2][2] = B[2][0], B[1][0], B[0][0]
        B[2][0], B[1][0], B[0][0] = tmp[0], tmp[1], tmp[2]

        return

def print_cube_upside():
    for i in range(3):
        string = ''
        for j in range(3):
            string += U[i][j]
        print(string)


def print_all(): #to debug
    for i in range(3):
        string = ''
        for j in range(3):
            string += U[i][j]
        print(string)

    for i in range(3):
        f = ''
        r = ''
        b = ''
        l = ''
        for j in range(3):
            f += F[i][j]
            r += R[i][j]
            b += B[i][j]
            l += L[i][j]
        print(f, r, b, l)

    for i in range(3):
        string = ''
        for j in range(3):
            string += D[i][j]
        print(string)

#t = 0
for test in range(int(input())):
    #t += 1
    #print('test case : ', t)
    init()
    n = int(input())
    order = input().split()
    index = 0
    for o in order:
        index += 1
        cnt = 3
        if o[1] == "+":
            cnt = 1
        for _ in range(cnt):
            rotate(o[0])

        #print(index)
        #print_all()
    print_cube_upside()

 

백준 5373 번 큐빙 자세한 설명 및 Python 코드였습니다.

감사합니다.

728x90
저작자표시 (새창열림)

'알고리즘 문제 풀이' 카테고리의 다른 글

124. [삼성기출20]Python 백준 16235 번 나무 재테크(시간 초과)  (0) 2023.06.08
123. [삼성기출19]Python 백준 16234 번 인구 이동 자세한 설명  (0) 2023.06.05
121. [삼성기출17]Python 백준 15686 번 치킨 배달 자세한 설명  (0) 2023.05.30
120. [삼성기출16]Python 백준 15685 번 드래곤 커브 자세한 설명  (0) 2023.05.26
119. [삼성기출15]Python 백준 15684 번 사다리조작 자세한 설명  (0) 2023.05.19
'알고리즘 문제 풀이' Related Articles more