[python]2016년

연습문제>2016년

[문제 설명]
2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT

입니다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 반환하세요.

[제한사항]

• 2016년은 윤년입니다.
• 2016년 a월 b일은 실제로 있는 날입니다.
  (13월 26일이나 2월 45일같은 날짜는 주어지지 않습니다)
  
  

[입출력 예]

나의 풀이

생각해보기!

def solution(a, b):
    a_b = [[0,0],[1,31],[2,29],[3,31],[4,30],[5,31],[6,30],
           [7,31],[8,31],[9,30],[10,31],[11,30],[12,31]]
    days = ['THU','FRI','SAT','SUN','MON','TUE','WED']
    cnt = 0
    
    for i in range(a):
        cnt += a_b[i][1]
        
    cnt = cnt + b
    answer = days[cnt%7]
    
    return answer

• 우선 1월부터 12월까지 [n월,n월 마지막 일] 리스트를 만들었다.

• 입력받은 월(a)의 전 달까지 마지막 일 수를 더하고 b 일수만큼 추가로 더해줍니다.

  • a = 3, b = 5 일 경우, 1월과 2월의 마지막 일 수(31,29)를 더한 후 5일(b)을 더해주면 3월 5일까지의 일 수를 알 수 있습니다.

• 전체 일 수를 7로 나눈 나머지에 따라 요일(days)을 반환하게 합니다.

  • 1월 1일이 금요일이므로 days 리스트의 인덱스가 1인 위치에 'FRI'를 두었습니다.

간단한 풀이지만 그 풀이를 생각하기까지 시간이 꽤 걸렸다.😫

다른 사람의 풀이 #1

def solution(a,b):
    
    answer = ""
    
    if a>=2:
        b+=31
        if a>=3:
            b+=29#2월
            if a>=4:
                b+=31#3월
                if a>=5:
                    b+=30#4월
                    if a>=6:
                        b+=31#5월
                        if a>=7:
                            b+=30#6월
                            if a>=8:
                                b+=31#7월
                                if a>=9:
                                    b+=31#8월
                                    if a>=10:
                                        b+=30#9월
                                        if a>=11:
                                            b+=31#10월
                                            if a==12:
                                                b+=30#11월
    b=b%7

    if b==1:answer="FRI"
    elif b==2:answer="SAT" 
    elif b==3:answer="SUN"
    elif b==4:answer="MON"
    elif b==5:answer="TUE"
    elif b==6:answer="WED"
    else:answer="THU"
        
    return answer

참고하세요!

• if문으로 해당하는 월(a)의 전 달까지 b에 계속해서 마지막 일 수를 더해줬다.
• 다 더한 b를 7로 나눠 나머지에 따라 요일을 지정했다.

다른 사람의 풀이 #2

def solution(a,b):
    months = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    days = ['FRI', 'SAT', 'SUN', 'MON', 'TUE', 'WED', 'THU']
    return days[(sum(months[:a-1])+b-1)%7]

참고하세요!

• 나의 풀이와 같은 맥락이지만 한 줄로 표현했다.👍