[코테준비] 삼성21. 코드트리: 정육면체 한번 더 굴리기 python & 삼성식 출력 & 삼성코테 언제부터 4시간?

728x90

이번 문제는 저번에 주사위 굴리기 문제를 푼 기억이 있어서 그런지 2시간 만에 푼 것 같다. 물론 골드 3이긴 하지만! 뿌듯 !

삼성코테 언제부터 4시간?

참고로 삼성 코테는 원래 3시간에 2문제였는데, 현재는 4시간에 2문제다. 2022 하반기부터 바뀌었다고 한다.

정육면체 한번 더 굴리기

링크: https://www.codetree.ai/training-field/frequent-problems/problems/cube-rounding-again/description
기출: 삼성 SW 역량테스트 2021 하반기 오전 1번 문제
난이도:골드 3
분류: 시뮬레이션
직접 푼 여부: O
언어: 파이썬

풀이 노트

정육면체 (=주사위)를 굴리는 문제를 안풀어봤다면, 백준의 주사위 굴리기 문제를 풀어보는 것을 추천한다!

[코테준비] 5. 백준- 주사위 굴리기(14499) python "삼성 SW 역량 테스트 기출 문제" (tistory.com)

그리고 이번에는 아래 노트에 나름 깔끔하게 나의 생각과정과 풀이를 적어 놓은 것 같다.

문제 이해 ( 사진이 잘 안보이면, 우클릭 후 "새 탭에서 이미지 열기"를 눌러 전체화면으로 보면 편하다!)

주의사항 (내가 놓친 부분)

(1) 또 방향 순서를 잘못 써서 틀렸다. "우 하 좌 상" 이였는데!! 아무생각없이 주사위를 굴릴 때, direcion =0,1,2,3 순서를 내맘대로 했다.

정답 코드 💻

# 정육면체 한번 더 굴리기
import sys
from collections import deque
# 기본 -------------------
input = sys.stdin.readline
# def -----------------
def in_range(r,c):
    if 0<=r<n and 0<=c<n:
        return True
    else:
        return False
def roll_dice(d, r,c):
    """
    주사위 굴리기 - 새 좌표, 주사위 전개도 수정
    :param d: 현재 주사위 굴리는 방향
    :param r, c: 원래 좌표
    :return: 새 좌표
    """
    global dice
    # [1] 새 좌표
    newr, newc = r+dr[d], c+dc[d]
    # [2] 주사위 전개도
    temp_dice = dice[:]
    if d==0:
        # 오른쪽
        dice=[temp_dice[left], temp_dice[right], temp_dice[top], temp_dice[bot], temp_dice[down], temp_dice[up]]
    elif d==1:
        # 아래
        dice = [temp_dice[up], temp_dice[down], temp_dice[right], temp_dice[left], temp_dice[top], temp_dice[bot]]

    elif d==2:
        # 왼쪽
        dice = [temp_dice[right], temp_dice[left], temp_dice[bot], temp_dice[top], temp_dice[down], temp_dice[up]]
    else:
        # 위
        dice=[temp_dice[down], temp_dice[up], temp_dice[right], temp_dice[left], temp_dice[bot], temp_dice[top]]

    return newr, newc

def point_cal(r,c):
    """
    점수 계산
    :param r: 좌표
    :param c: 좌표
    :return: 이번에 얻은 점수
    """
    pointnum = grid[r][c]
    point = grid[r][c]
    visited =[[False]*n for _ in range(n)]
    visited[r][c] =True
    q =deque()
    q.append((r,c))
    while q:
        r, c = q.popleft()
        for i in range(4):
            newr, newc = r+dr[i], c+dc[i]
            if in_range(newr,newc) and not visited[newr][newc] and grid[newr][newc] ==pointnum:
                visited[newr][newc] = True
                point +=pointnum
                q.append((newr, newc))

    return point

def change_direction(d,r,c):
    """
    주사위 굴러갈 방향 change
    :param d: 현재 방향
    :param r: 현재 좌표
    :param c: 
    :return: 새로운 방향 (가능한 곳으로) 0~3
    """
    # [1] 주사위와 grid 비교
    dice_bottom = dice[bot]
    grid_now = grid[r][c]
    if dice_bottom> grid_now:
        #시계방향
        newdir = (d+1)%4
        newr, newc = r+dr[newdir], c+dc[newdir]
    elif dice_bottom<grid_now:
        # 반시계
        newdir = (d+3)%4
        newr, newc = r+dr[newdir], c+dc[newdir]
    else:
        #같으면, 그대로
        newdir = d
        newr, newc = r+dr[newdir], c+dc[newdir]

    # [2] 만약 not in range()
    if not in_range(newr, newc):
        newdir = (newdir+2)%4
    return newdir
# prepare ----------------
# row, column 이동
dr = [0,1,0,-1] # 우 하 좌 상 순
dc = [1,0,-1,0]
# 주사위 위치 변경 편하게 (index) - top, bottom이 윗면 아랫면, 윗면기준으로 동서남북 순으로 오왼하상
top, bot, right, left, down, up= 0,1,2,3,4,5
# 초기 방향
direction =0 # 오른쪽
# 초기 주사위
dice = [1,6,3,4,2,5]
# 주사위 초기 위치
r, c = 0,0
# 총 점수
total_point =0
# input -----------------
n, m = map(int, input().split())
grid = [list(map(int, input().split())) for _ in range(n)]

# main --------------------
for step in range(m):
    # [1] 주사위 굴리기
    r, c = roll_dice(direction,r,c) #주사위 위치 in grid
    # [2] 점수 계산
    total_point += point_cal(r, c)
    # [3] 방향 바꾸기 (새로운 위치)
    direction = change_direction(direction,r,c)

print(total_point)

정답 코드 (삼성st)💻

저 예시 출력이 나오는 예시 input

그리고 코드

# 정육면체 한번 더 굴리기
import sys
from collections import deque
# 기본 -------------------
#sys.stdin = open("./input.txt")
#input = sys.stdin.readline
# def -----------------
def in_range(r,c):
    if 0<=r<n and 0<=c<n:
        return True
    else:
        return False
def roll_dice(d, r,c):
    """
    주사위 굴리기 - 새 좌표, 주사위 전개도 수정
    :param d: 현재 주사위 굴리는 방향
    :param r, c: 원래 좌표
    :return: 새 좌표
    """
    global dice
    # [1] 새 좌표
    newr, newc = r+dr[d], c+dc[d]
    # [2] 주사위 전개도
    temp_dice = dice[:]
    if d==0:
        # 오른쪽
        dice=[temp_dice[left], temp_dice[right], temp_dice[top], temp_dice[bot], temp_dice[down], temp_dice[up]]
    elif d==1:
        # 아래
        dice = [temp_dice[up], temp_dice[down], temp_dice[right], temp_dice[left], temp_dice[top], temp_dice[bot]]

    elif d==2:
        # 왼쪽
        dice = [temp_dice[right], temp_dice[left], temp_dice[bot], temp_dice[top], temp_dice[down], temp_dice[up]]
    else:
        # 위
        dice=[temp_dice[down], temp_dice[up], temp_dice[right], temp_dice[left], temp_dice[bot], temp_dice[top]]

    return newr, newc

def point_cal(r,c):
    """
    점수 계산
    :param r: 좌표
    :param c: 좌표
    :return: 이번에 얻은 점수
    """
    pointnum = grid[r][c]
    point = grid[r][c]
    visited =[[False]*n for _ in range(n)]
    visited[r][c] =True
    q =deque()
    q.append((r,c))
    while q:
        r, c = q.popleft()
        for i in range(4):
            newr, newc = r+dr[i], c+dc[i]
            if in_range(newr,newc) and not visited[newr][newc] and grid[newr][newc] ==pointnum:
                visited[newr][newc] = True
                point +=pointnum
                q.append((newr, newc))

    return point

def change_direction(d,r,c):
    """
    주사위 굴러갈 방향 change
    :param d: 현재 방향
    :param r: 현재 좌표
    :param c: 
    :return: 새로운 방향 (가능한 곳으로) 0~3
    """
    # [1] 주사위와 grid 비교
    dice_bottom = dice[bot]
    grid_now = grid[r][c]
    if dice_bottom> grid_now:
        #시계방향
        newdir = (d+1)%4
        newr, newc = r+dr[newdir], c+dc[newdir]
    elif dice_bottom<grid_now:
        # 반시계
        newdir = (d+3)%4
        newr, newc = r+dr[newdir], c+dc[newdir]
    else:
        #같으면, 그대로
        newdir = d
        newr, newc = r+dr[newdir], c+dc[newdir]

    # [2] 만약 not in range()
    if not in_range(newr, newc):
        newdir = (newdir+2)%4
    return newdir
# prepare ----------------
# row, column 이동
dr = [0,1,0,-1] # 우 하 좌 상 순
dc = [1,0,-1,0]
# 주사위 위치 변경 편하게 (index) - top, bottom이 윗면 아랫면, 윗면기준으로 동서남북 순으로 오왼하상
top, bot, right, left, down, up= 0,1,2,3,4,5


# 전체 main --------------------
total_test = int(input())
for round in range(total_test):
    # 초기화 ----------------
    # 초기 방향
    direction = 0  # 오른쪽
    # 초기 주사위
    dice = [1, 6, 3, 4, 2, 5]
    # 주사위 초기 위치
    r, c = 0, 0
    # 총 점수
    total_point = 0
    # input -----------------
    n, m = map(int, input().split())
    grid = [list(map(int, input().split())) for _ in range(n)]
    # main 문
    for step in range(m):
        # [1] 주사위 굴리기
        r, c = roll_dice(direction, r, c)  # 주사위 위치 in grid
        # [2] 점수 계산
        total_point += point_cal(r, c)
        # [3] 방향 바꾸기 (새로운 위치)
        direction = change_direction(direction, r, c)

    print(f"#{round+1} {total_point}")

728x90

'취준! ✒ > 삼성' 카테고리의 다른 글

[코테준비] 삼성20. 코드트리: 꼬리잡기놀이 python & 삼성식 출력 (0)	2023.10.12
[코테준비] 삼성19. 삼성 코테 준비전에 꼭 볼 것- SW Expert에 대해서! (feat. 보호 필름 문제 python) (1)	2023.10.11
[코테준비] 삼성18. 코드트리: 싸움땅 python (1)	2023.10.10
[코테준비] 삼성17. 코드트리: 코드트리 빵 python + 코드트리의 채점 무한로딩 (1)	2023.10.09
[코테준비] 삼성16. 코드트리: 포탑 부수기 python (1)	2023.10.07

취준생 메모장 (AI & CV engineer) 💻

[코테준비] 삼성21. 코드트리: 정육면체 한번 더 굴리기 python & 삼성식 출력 & 삼성코테 언제부터 4시간?

삼성코테 언제부터 4시간?

정육면체 한번 더 굴리기

풀이 노트

주의사항 (내가 놓친 부분)

정답 코드 💻

정답 코드 (삼성st)💻

'취준! ✒ > 삼성' 카테고리의 다른 글

티스토리툴바

[코테준비] 삼성21. 코드트리: 정육면체 한번 더 굴리기 python & 삼성식 출력 & 삼성코테 언제부터 4시간?

삼성코테 언제부터 4시간?

정육면체 한번 더 굴리기

풀이 노트

주의사항 (내가 놓친 부분)

정답 코드 💻

정답 코드 (삼성st)💻

'취준! ✒ > 삼성' 카테고리의 다른 글

관련글

티스토리툴바