[프로그래머스] 2016년

programmers.co.kr/learn/courses/30/lessons/12901?language=cpp

코딩테스트 연습 - 2016년

---

문제풀이)

풀기쉽게 월화수목금토일을 저장해주는 day배열(1월1일은 금요일이니 금,토,일,월,화,수,목 순으로 저장)을 만들고,

월별 며칠까지 있는지를 month배열에 저장해둔다. 윤년이므로 2월달은 29일로 저장해준다.

 

그런뒤 a월 이전까지의 총 날짜를 다 더해주고, 남은b일을 더한다음 1을 빼주고 modulo(%) 7을 해준다음 해당 요일의 값을 return 해주면 된다.

 

#include <string>
#include <vector>

using namespace std;
string day[7]={"FRI","SAT","SUN","MON","TUE","WED","THU"};
int month[12]={31,29,31,30,31,30,31,31,30,31,30,31};
string solution(int a, int b) {
    string answer = "";
    int sum=-1;
    for(int i=1;i<a;i++){
        sum+=month[i-1];
    }
    sum+=b;
    answer=day[sum%7];
    return answer;
}