[프로그래머스] 2016년

https://programmers.co.kr/learn/courses/30/lessons/12901

코딩테스트 연습 - 2016년

2016년 1월 1일은 금요일이므로,

2016 년 A월 B일 은 A - 1 월까지의 일 수 + B 를 더한 값 data 를 7로 나눈 나머지에 따라서 요일을 알 수 있다.

 

[C++]

#include <string>
#include <vector>

using namespace std;

/*
(1, 31) (2,29) (3, 31) (4, 30) (5, 31)
(6, 30) (7, 31) (8, 31) (9, 30) (10, 31)
(11, 30) (12,31)
*/

//2016. 01. 01 : FRI

string solution(int a, int b) {
    string answer = "";
    int date = 0;

    int day_arr[] = { 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
    string arr[] = { "FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU" };

    for (int i = 0; i < a-1; i++)
    {
        date += day_arr[i];
    }
    date += b;

    if (date % 7 - 1 != -1)
        answer = arr[date % 7 - 1];
    else // date % 7 == 0 이면 arr[-1] 이 되는 것에 대한 예외처리
        answer = arr[6]; 

    return answer;
}