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

 

 

 

switch에서 바로 return을 줘서 간단하게 구현해봤다.

 

#include <string>
#include <vector>

using namespace std;

string solution(int a, int b) {
    int day = b-1;
	for (int i = 1; i < a; i++) {
		if (i == 2) day += 29;
		else if (i == 4 || i == 6 || i == 9 || i == 11) day += 30;
		else day += 31;
	}
	switch (day%7)
	{
	case 0:
		return "FRI";
	case 1:
		return "SAT";
	case 2:
		return "SUN";
	case 3:
		return "MON";
	case 4:
		return "TUE";
	case 5:
		return "WED";
	case 6:
		return "THU";
	default:
		break;
	}
}