[프로그래머스] 2016 with Swift

문제 요약

2016년 a월 b일이 무슨 요일인지 리턴하는 함수 작성

2016년은 윤년 

2016년 a월 b일은 실제로 있는 날

 

접근

00. a월 b일이 총 몇 일째인지 구해서 7로 나누자

01. 2016년 1월 1일이 무슨 날? - 금요일

-> 1월 1일 -> 총 하루 -> 7을 1로 나눈 나머지 즉, 1일 때 금요일

-> 1 - 금, 2 - 토, 3 - 일, ... , -목

 

코드

func solution(_ a:Int, _ b:Int) -> String {
    
    var answer: String = ""
    var totalDay: Int = b
    var monthDay: Int = 0
    
    
    for index in 1..<a {
        
        if ( index < 8 && index % 2 == 1 ) || ( index >= 8 && index % 2 == 0 ){
            monthDay = 31
        }
        else if index == 2 {
            monthDay = 29
        } else {
            monthDay = 30
        }
        
        totalDay += monthDay
    }
    
    switch totalDay % 7 {
    case 0:
        answer = "THU"
        break;
    case 1:
        answer = "FRI"
        break;
    case 2:
        answer = "SAT"
        break;
    case 3:
        answer = "SUN"
        break;
    case 4:
        answer = "MON"
        break;
    case 5:
        answer = "TUE"
        break;
    case 6:
        answer = "WED"
        break;
    default:
        answer = ""
    }
    
    
    return answer
}