[파이썬] 백준 알고리즘_1924번

문제: 오늘은 2007년 1월 1일 월요일이다. 그렇다면 2007년 x월 y일은 무슨 요일일까? 이를 알아내는 프로그램을 작성하시오.

입력 : 첫째 줄에 빈 칸을 사이에 두고 x(1≤x≤12)와 y(1≤y≤31)이 주어진다. 참고로 2007년에는 1, 3, 5, 7, 8, 10, 12월은 31일까지, 4, 6, 9, 11월은 30일까지, 2월은 28일까지 있다.

1 1

출력 : 첫째 줄에 x월 y일이 무슨 요일인지에 따라 SUN, MON, TUE, WED, THU, FRI, SAT중 하나를 출력한다.

MON

내 소스

a=input().split()

b=int(a[0])

d =[]

for c in range(1, b):

    if c == 2:

        d.append(28)

    elif c == 1 or c==3 or c==5 or c==7 or c==8 or c==10 or c==12:

        d.append(31)

    elif c == 4 or c==6 or c==9 or c==11:

        d.append(30)

    else:

        print('오류')

e=(sum(d)+int(a[1]))%7

if e==1:

    print('MON')

elif e==2:

    print('TUE')

elif e ==3:

    print('WED')

elif e ==4:

    print('THU')

elif e ==5:

    print('FRI')

elif e ==6:

    print('SAT')

else:

    print('SUN')