프로그래머스 - 2016년

프로그래머스 - 2016년

Level 1로 쉬운 문제이다.
7월을 기준으로 홀수 달과 짝수 달의 날짜가 달라지는 것만 주의하면 된다.

#include <string>
#include <vector>

using namespace std;

string solution(int a, int b) {
    string answer = "";
    vector<string> day = { "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"};
    int mon = 0;
    for(int i=1;i<a;i++) { 
        if(i == 2) {
            mon += 29;
            continue;
        }
        if(i <= 7) 
            mon += (i % 2 == 1) ? 31 : 30;
        else 
            mon += (i % 2 == 1) ? 30 : 31;
    }
    int date = mon + b + 4;
    answer = day[date % 7];
    return answer;
}