45] 프로그래머스 2016년

코딩테스트 연습

 

각 배열 0번째값 조심

0, THU.

#include <string>
using namespace std;

string solution(int a, int b) {
    string answer = "";
    int arr[13] = {0,31, 29,31,30,31,30,31,31,30,31,30,31};
    string date[7] = { "THU", "FRI", "SAT", "SUN", "MON" ,"TUE", "WED"};
    int cnt = 0;
    for(int i = 0;i<a;i++)
        cnt+=arr[i];
    cnt += b;
    
    answer = date[cnt%7];
    
    return answer;
}