프로그래머스(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;
}
반응형