Swift랑 친해지기/programmers 풀기

[프로그래머스] 2016년 (Swift)

데브킹덕 2022. 7. 25. 16:51

https://school.programmers.co.kr/learn/courses/30/lessons/12901

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

1. dayOfTheWeek 상수배열생성 

2. lastday 1월~12월까지 마지막날짜 상수배열 생성

3. 1월 1일 금요일 이므로 dayOfTheWeek에 인덱스값 5 (인덱스는 0부터 시작)

4. for loop를 이용해 인덱스 0부터 a-1까지 마지막날을 모두 더함 

이때 a-1 의 이유는 원하는 a 달의 전달의 마지막날까지 더하고 b를 더하기 위함 

5. firstdayIndex에 b-1을 더함 

이유는 b-1을 더하기 직전상황이 a달에 1일이기 때문임

6. firstdayIndex 를 7로 나눈 나머지를 dayOfTheWeek에 인덱스로 반환하면 원하는 요일이 나옴 

 

func solution(_ a:Int, _ b:Int) -> String {
    let dayOfTheWeek = ["SUN","MON","TUE","WED","THU","FRI","SAT"]
    let lastday = [31,29,31,30,31,30,31,31,30,31,30,31]
    
    var firstdayIndex = 5
    
    for i in 0 ..< a-1 {
        firstdayIndex += lastday[i]
    } 
    
    firstdayIndex += b-1
    
    return "\(dayOfTheWeek[firstdayIndex%7])"
}