[프로그래머스] 2016년 _ C++

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

프로그래머스

 

흔히 나오는 윤년을 알고있기만 하면

덧셈으로 편하게 풀 수 있다

해당 2016년만 확인하면 되기에,

각 달이 몇일로 구성되어 있는지 배열에 넣어주고,

1월 1일로 시작하는 요일부터 배열에 넣어주고,

1월 1일로부터 몇일 지났는지 확인 한 다음 7로 나누어주면 요일을 확인할 수 있다.

#include <string>
#include <vector>
 
using namespace std;
 
string solution(int a, int b) {
    string answer = "";
    
    int month[] = {0,31,29,31,30,31,30,31,31,30,31,30,31};
    string day[] = {"FRI","SAT","SUN","MON","TUE","WED","THU"};
    
    int num = b-1; // 1일로부터 b-1일이 지난 후
    for (int i = 0; i < a; i++)
    {
        num += month[i];
    }
    int mod = num%7;
    
    return day[mod];
}