일주일은 7일이고 1월1일부터 날짜를 세서 7로 나눈 나머지를 요일로 선택하게 하는 방법으로 하기로 했다.

이때 1월1일이 금요일이라서 요일 배열의 1번째 값을 금요일로 해두었다.

#include <string>
#include <vector>

using namespace std;

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

+ Recent posts