백준 1924 파이썬

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

1924번: 2007년

구해야 하는 바는 간단하다! 2007년 x월 y일이 무슨 날짜인지 구하면 된다. 

2007년 1월 1일은 월요일이다. 

 

나는 이 문제를 어떻게 접근해야 간단하게 할 수 있을지 고민을 꽤 한 것 같다. 처음에 고민한 것은 월별로 31일이거나, 30일이거나, 28일이 되는데 이걸 어떻게 계산해야 그 다음 월의 시작 요일을 알 수 있을까, 하는 것이었다. 같은 달 내에서는 7일마다 반복되는 규칙이 생기지만 월이 바뀌면 그 규칙이 모두 바뀌어 버리기 때문이다.

 

내가 생각한 방식은 큐를 만드는 것이었다.

 

dates = deque(['SUN', 'MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT'])

큐에 sun을 0번째 인덱스로 넣고 차례대로 sat까지 출력하고자 하는 문자열을 넣어준다.

우선 처음 시작할 때 1월 1일은 월요일이다. 즉, dates라는 큐의 첫번째 인덱스(0번째 인덱스부터 시작하므로 첫번째 인덱스는 MON이 맞음)와 일치한다.

 

1월 2일은 7%2 == 2. 즉 두 번째 인덱스와 일치한다.

 

인덱스와 계속해서 일치한 결과를 출력하고 싶다면 어떤 특정한 월의 1일이 나의 첫번째 인덱스가 되면 되겠다는 생각이 들었다.

 

즉, 만약 3월 1일이 수요일이라면 수요일이 dates[1]의 값이여야 한다. 이렇게 맞추어 주면 3월의 어떤 y일이 입력되어도 y % 7의 값은 규칙에 따라 수, 목, 금, 토, 일, 월, 화 형식을 따르게 되어 있다. 그 형식을 맞출 수 있도록 큐의 값을 pop하고 append해주면 되는 것이다.

 

그리고 그 규칙을 만들기 위해 조심해야 하는 것이 월마다 날짜가 31일, 30일, 28일로 다르다는 것이다.

 

결론적으로 만들어진 for 문은

zeroIndex = 1

for month in range(1, x):
    if month==2: zeroIndex = 28%7
    elif month ==4 or month==6 or month==9 or month==11: zeroIndex = 30%7
    else: zeroIndex = 31%7
    for i in range(zeroIndex):
        dates.append(dates.popleft())

즉, month의 range(1,x) (여기서 x는 입력받은 x월) 동안 case를 나누어서 계산을 실행한다. 여기서 주의해야 할 점은 만약 우리가 3월 3일의 결과를 알고 싶다면 x == 3, 즉 for문은 month가 1, 2 일때만 돌게 된다는 것이다. 즉, 내가 for문에서 구하고자 하는 것은 3월 직전달인 2월달까지의 결과이다. 왜냐하면 for문 안에 중첩되어 있는 for 문(range(zeroIndex)인)에서 구해주는 것이 2월달 날짜에서의 규칙을 깨고 새롭게 3월달의 규칙을 만들기 위해서 정의된 것이기 때문이다.

 

month가 1일 때는 else에 들어가서 zeroIndex 값이 3이 된다. 즉, 1월 31일이 31 % 7 == 3으로 dates[3]의 'WED'가 된다는 뜻이다. 그리고 2월 1일이 'THU'가 되어야 한다는 뜻이기도 하다. 즉, dates의 첫 번째 인덱스가 'THU'로 바뀌어야 한다는 뜻이다. 

 

그래서 for문을 다시 돌면서 dates.popleft()를 3번 해주면 SUN, MON, TUE 가 pop되고 다시 dates에 append되어,

dates = ['WED', 'THU', 'FRI', 'SAT', 'SUN', 'MON', 'TUE'] 형식으로 바뀌게 되는 것이다.

 

그러면 2월에서는 y일 % 7의 값이 0이면 수요일, 1이면 목요일, 2면 금요일... 이런식의 패턴이 생성되는 것이다.

 

이게 3월달에도 그대로 적용된다면 2월의 마지막 날짜는 28 % 7 == 0: 즉 수요일이다. 그럼 3월 1일은 목요일이 되고, 다시 2월달에서 했던 것처럼 Pop과 append를 반복해주면 된다.

 

그리고 마지막에 연산이 끝나고 내가 원하는 값을 출력하기 위해서는 print(dates[y%7])을 해주면 된다.

 

<전체 코드>

from sys import stdin
from collections import deque

x, y = map(int, stdin.readline().split())
zeroIndex = 1
dates = deque(['SUN', 'MON', 'TUE', 'WED', 'THU', 'FRI', 'SAT'])

for month in range(1, x):
    if month==2: zeroIndex = 28%7
    elif month ==4 or month==6 or month==9 or month==11: zeroIndex = 30%7
    else: zeroIndex = 31%7
    for i in range(zeroIndex):
        dates.append(dates.popleft())

print(dates[y%7])