[백준] 9655. 돌 게임/Python - Silver5

❓문제

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

성능 요약

메모리: 32544 KB, 시간: 36 ms

분류

수학, 다이나믹 프로그래밍, 게임 이론

 

문제 설명

돌 게임은 두 명이서 즐기는 재밌는 게임이다.

탁자 위에 돌 N개가 있다. 상근이와 창영이는 턴을 번갈아가면서 돌을 가져가며, 돌은 1개 또는 3개 가져갈 수 있다. 마지막 돌을 가져가는 사람이 게임을 이기게 된다.

두 사람이 완벽하게 게임을 했을 때, 이기는 사람을 구하는 프로그램을 작성하시오. 게임은 상근이가 먼저 시작한다.

✍🏻풀이

N = 1 일 때는 상근이가 이긴다.
N = 2 일 때는 창영이가 이긴다.
N = 3 일 때는 상근이가 3개를 모두 가져가거나 1 개씩 가져가서 상근➡️창영➡️상근으로 상근이가 이긴다.

이 상황으로 봤을 때 N인 홀수일 때는 상근이가 이기고 짝수일 때는 창영이가 이긴다는 것을 알 수 있다.

💻코드

내 풀이

import sys

input = sys.stdin.readline

N = int(input())

print("SK" if N%2 == 1 else "CY")

DP 사용한 풀이

import sys

input = sys.stdin.readline

N = int(input())
dp = [-1] * 1001
dp[1] = 1
dp[2] = 0
dp[3] = 1

for i in range(4, N+1):
    if dp[i-1] != 1 or dp[i-3] != 1:
        dp[i] = 1
    else:
        dp[i] = 0

print("SK" if dp[N] == 1 else "CY")

📍후기

N이 홀수인 경우에는 무조건 상근이가 이기고 짝수인 경우에는 창영이가 이기는 걸 바로 확인하고 수학으로 풀었는데 DP로 풀이로도 풀 수 있다니 신기했다.

그리고 이 문제는 PyPy3보다 Python으로 컴파일했을 때가 더 빠르다.