프로그래머스 Lv.1 코딩테스트

[프로그래머스 Lv.1] 2016년

하다블 2022. 3. 2. 19:33
반응형

문제는 다음과 같습니다.

코딩테스트 연습 - 2016년 | 프로그래머스 (programmers.co.kr)

 

코딩테스트 연습 - 2016년

2016년 1월 1일은 금요일입니다. 2016년 a월 b일은 무슨 요일일까요? 두 수 a ,b를 입력받아 2016년 a월 b일이 무슨 요일인지 리턴하는 함수, solution을 완성하세요. 요일의 이름은 일요일부터 토요일까

programmers.co.kr

 

풀이 코드는 다음과 같습니다.

#include <string>
#include <vector>

using namespace std;

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

한 달이 지날 때 마다 지난 달이 몇일이나 있는지 확인해주기 위해 배열을 사용하였고 다음 달이 되면 0일부터 시작하지 않고 1일부터 시작하기 때문에 이러한 부분만 신경써주면 풀 수 있을 것으로 보입니다.

반응형