쌓고 쌓다

[프로그래머스] 2016년 C++ 풀이 본문

알고리즘/프로그래머스

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

승민아 2022. 6. 30. 23:52

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

 

코딩테스트 연습 - 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 = "";
    string day[7] = {"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};
    int MON[13] = {0 ,31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    int cnt=0;
    for(int i=1;i<a;i++)
        cnt+=MON[i];
    cnt+=b-1;
    answer+=day[cnt%7];
    
    return answer;
}

 

    cnt+=b; 가 아닌 cnt+=b-1; 인 이유는 

이 부분은 1일부터 시작하기에 이미 날짜 b에 1은 포함 되어 있기 때문입니다.

예를 들어 1월 1일이 입력으로 들어왔다고 합시다.

b가 1인 경우 cnt에 1을 더해버리는데 그러면

cnt 에 1이 들어오고 day[1]을 출력하는데 엉뚱하게 SAT가 출력이 됩니다.

day를 시작할때 0 번째 인덱스는 FRI입니다.

 

Comments