갓비니

programmers 연습문제 : 2016년 / C++ 코드 본문

Programming/코딩 문제 풀이

programmers 연습문제 : 2016년 / C++ 코드

갓비니 2020. 7. 3. 00:53

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
#include <string>
#include <vector>
 
using namespace std;
 
string solution(int a, int b) {
    string DAY[7= { "THU""FRI","SAT","SUN","MON","TUE","WED" };
    int day = 0;
    int sum = 0;
    string answer = {};
    if ((a == 1&& (b <= 7))answer = DAY[b % 7];
    else {
        for (int i = a - 1; i>0; i--) {
            if ((i == 1|| (i == 3|| (i == 5|| (i == 7|| (i == 8|| (i == 10|| (i == 12))day = 31;
            else if (i == 2)day = 29;
            else day = 30;
            sum = sum + day;
        }
        sum = sum + b;
 
 
        answer = DAY[sum % 7];
    }
 
    return answer;
}
cs