euisblue
12901. 2016년

Level 1 2016년

시간 복잡도: O(M), M = Month (1~12)

1#include <string> 2#include <vector> 3 4using namespace std; 5 6string solution(int a, int b) { 7 int days[] = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31}; 8 string day[] = {"THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"}; 9 int nDays = 0; 10 11 for(int i=0; i<a-1; ++i) nDays += days[i]; 12 nDays += b; 13 14 return day[nDays%7]; 15}

1월 1일이 월요일이 아닌 목요일이라고 했기 때문에 배열을 THU에서 시작했다. 만약 MON부터 시작을 했다면 마지막에 +3의 인덱스를 반환해주면 된다.