[Programmers - lv01] 2016년 (cpp / python)

2016년

코딩테스트 연습 - 2016년

문제 설명

주어진 a월 b까지의 일 수를 모두 계산하여 7로 나눈 나머지를 이용하여 탐색

문제 풀이

*소스코드 : C++ *

#include <iostream>
#include <string>
#include <vector>

using namespace std;

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

소스코드 : Python

def solution(a, b):
    day = ["FRI","SAT","SUN","MON","TUE","WED","THU"]
    date_of_month = [31,29,31,30,31,30,31,31,30,31,30,31]
    return day[(sum([ date_of_month[i] for i in range(a-1)]) + b -1)%7]