[백준] 9657번: 돌 게임 3 - python

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

9657번: 돌 게임 3

 

저번에 처음 돌 게임 1을 푼 기억이 생각났다.

 

확실히 업그레이드가 되었다.

가지치기를 하면 훨씬 빠르지만 게을러서 그냥 제출헀다.

n = int(input())
dp = [0] * 1001
#0은 상근 1은 창영으로 설정
dp[1] = 0
dp[2] = 1
dp[3] = 0
dp[4] = 0
dp[5] = 0
dp[6] = 0 # 4 1 1
dp[7] = 1 # 1 4 1 1
for i in range(8, 1001):
    if dp[i-1] == 1 or dp[i-3] == 1 or dp[i-4] == 1:
        dp[i] = 0
    else:
        dp[i] = 1

if dp[n] == 0:
    print("SK")
else:
    print("CY")

 

완성