[프로그래머스 / C++] 2016년

문제

2016년 : 문제 링크

문제 분석

• 2016년 1월 1일은 금요일이다. 2016년 a월 b일은 무슨 요일일까? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성. 요일의 이름은 일요일부터 토요일까지 각각 SUN,MON,TUE,WED,THU,FRI,SAT 이다. 예를 들어 a=5, b=24라면 5월 24일은 화요일이므로 문자열 "TUE"를 return

• 제한 사항

• 2016년은 윤년이다.
• 2016년 a월 b일은 실제로 있는 날이다. (13월 26일이나 2월 45일같은 날짜는 주어지지 않는다)

• 1월 1일부터 주어진 날짜까지 총일수를 저장할 정수형 변수 day를 0으로 초기화. 문자열형 벡터 dow에는 1월 1일인 "FRI"(금요일)부터 "THU"(목요일까지 순서대로 저장하고, 정수형 벡터 days에는 1월부터 12까지의 일수를 모두 저장(2016은 윤년이므로 2월은 29일까지 있음). for loop를 통해 a - 1월 까지의 일수를 day에 더하고 저장하고, loop 탈출 후 b를 day에 더하고 저장. day에서 1을 뺀값(dow의 첫번째 인덱스는 0이므로)에서 7을 나눈 나머지가 인덱스가 되고, dow에서 해당 인덱스에 저장된 문자열을 return

풀이

#include <string>
#include <vector>

using namespace std;

string solution(int a, int b) {
    int day = 0;
    vector<string> dow{"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};
    vector<int> days{31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    
    for(int i = 0; i < a - 1; ++i) day += days[i];
    day += b;
    return dow[(day - 1) % 7];
}