프로그래머스(Programmers)

프로그래머스 2016년 C++

cosmohoo 2022. 2. 4. 16:25
반응형

문제 설명

 

=> 단순구현 문제입니다. 

=> 풀이는 아래의 그림을 참조해주시기 바랍니다. 

 

 

 

#include <string>
#include <vector>

using namespace std;

vector<string> seven_day = {"FRI","SAT","SUN","MON","TUE","WED","THU"};
vector<int> arr ={0,31,29,31,30,31,30,31,31,30,31,30,31};

string solution(int a, int b) {
    string answer = "";
    int day=0; 
    for(int i=1; i<a; i++)
    {
        day += arr[i];
    }
    
    day = day + (b-1);
    answer = seven_day[day%7];
    
    return answer;
}
반응형