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

요약

• 날짜(월, 일)을 인자로 받아 요일을 반환
• 조건: 2016년 1월 1일은 금요일, 2016년은 윤년

풀이

내 풀이

#include <string>
#include <vector>

using namespace std;

string solution(int a, int b) {
  vector<string> days = {"THU", "FRI", "SAT", "SUN", "MON", "TUE", "WED"};
  vector<int> lens = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
  int dayLen = b;

  for (int i = 0; i < (a-1); i++) {
    dayLen += lens[i];
  }
  return days[dayLen % 7];
}

int main() {
  assert(solution(5, 24) == "TUE");
}