[C++] 프로그래머스 레벨 1 - 2016년

programmers.co.kr/learn/courses/30/lessons/12901?language=cpp

코딩테스트 연습 - 2016년

 

2016년은 윤년이다.

4로 나누어떨어지면 윤년.

4로 나누어떨어져도 100으로 떨어지면 평년,

4로 나누어떨어지고, 100으로 나누어 떨어지면서, 400으로 나누어떨어지면 윤년이다.

 

날짜 계산만 잘 하고, b는 금요일부터 시작했기 때문에 4를 더해주었다. week 벡터에는 0번 시작이 SUN이기 떄문에

 

#include <string>
#include <vector>
#include <map>

using namespace std;

string solution(int a, int b) {
    string answer = "";
    int day = 0;
    
    map<int, int> year;
    year[1] = 31;
    year[2] = 29;
    year[3] = 31;
    year[4] = 30;
    year[5] = 31;
    year[6] = 30;
    year[7] = 31;
    year[8] = 31;
    year[9] = 30;
    year[10] = 31;
    year[11] = 30;
    year[12] = 31;

    vector<string> week = {"SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"};
    
    for(int i = 1; i < a ; i++){
        day += year[i];
    }
    day += b + 4;
    
    answer = week[day%7];
    
    return answer;
}