[프로그래머스] 레벨 1-1번

 

1월 1일이 금요일이라 할 때 a월 b일이 무슨 요일인지 출력하는 문제였다.

2016년이라 2월이 윤달이라고 했다.

---

12월 각각의 일수를 기록한 후 목, 금, ... , 수요일까지 저장한다.

a월 전까지 모든 일수를 더한 후 b일도 더해서 7로 mod 계산을 한다.

나머지가 0이면 목요일 1이면 금요일, 6이면 수요일을 출력한다.

어렵지않은 기본 문제이다.

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/Programmers/%EB%A0%88%EB%B2%A81_1%EB%B2%88.cpp

 

C++ 코드

#include <string>
#include <iostream>
#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 };
	vector<string> res = { "THU", "FRI","SAT","SUN","MON","TUE","WED" };
	int day = b;
	for (int i = 1; i < a; i++)
		day += month[i];
	day %= 7;
	answer = res[day];
	return answer;
}

int main()
{
	string tmp = solution(1, 1);
	cout << tmp;
}