프로그래머스 - 2016년 / C++

문제 링크 : programmers.co.kr/learn/courses/30/lessons/12901

코딩테스트 연습 - 2016년

2016년의 a월 b일을 입력받아 요일을 출력하는 문제이다.

2016년은 윤년이므로 

1 -> 31일
2 -> 29일
3 -> 31일
4 -> 30일
5 -> 31일
6 -> 30일
7 -> 31일
8 -> 31일
9 -> 30일
10 -> 31일
11 -> 30일
12 -> 31일

까지 있다. 총 366일이다.

 

아이디어는 지난달의 일 수를 모두 더하고, 이번 달의 일 수를 더해서 7로 나눈 나머지로 요일을 정하는 것이다.

나머지가 0이라면 금요일, 1이라면 토요일, 2라면 일요일 .. 로 나오게 된다.

 

만약 1월 3일이라면 지난 달이 없으므로 0을 더하고, 3일을 더하는데 일 수를 더할 때는 1일을 포함하므로 1을 빼준다.

따라서 나머지가 2이므로 SUN(일요일)이 출력된다.

 

<소스 코드>

#include <iostream>
#include <string>
#include <vector>
using namespace std;

string solution(int a, int b) {
    string answer = "";
    int day = 0;
    int m[13] = {0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    string weeks[7] = {"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};
    
    for(int i=1; i <= a; i++){
        day += m[i-1];
    }
    day += b-1;
    
    answer = weeks[day%7];
    
    return answer;
}