45] 프로그래머스 2016년

IT/알고리즘

45] 프로그래머스 2016년

깻잎쌈 2020. 9. 15. 21:01

코딩테스트 연습

기초부터 차근차근, 직접 코드를 작성해 보세요.

programmers.co.kr

각 배열 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;
}