[프로그래머스 / Level 1] 2016년

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

코딩테스트 연습 - 2016년

문제 접근법

• 다른 년도를 신경 쓸 필요없이 2016년만 신경쓰면 되는 문제입니다.
• 요일에 대한 string 배열(저는 answer로 표현하겠습니다)과 1월~12월에 해당하는 일수에 대한 int 배열을 준비합니다(저는 month로 표현하겠습니다).
• for문을 통해 a월 이전까지의 month배열값을 더해줍니다(이 값을 num이라 표현하겠습니다).
• num값과 b를 더하고 -1을 한값을 나머지 연산을 합니다(어떤 요일인지를 구하기 위하여)
  • -1을 한 이유는 answer 배열의 index가 0부터 시작하기 때문입니다.
• 해당 값을 answer배열의 index로 주고 return 하면 됩니다.

아래는 코드입니다.

#include <string>
#include <vector>
 
using namespace std;
 
string solution(int a, int b) {
    string answer[] = {"FRI", "SAT", "SUN", "MON", "TUE", "WED", "THU"};
    int month[] = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    int count = 0;
    
    for(int i = 0; i < a - 1; i++)
        count += month[i];
    
    count += b - 1;
    count %= 7;
    
    return answer[count];
}