[프로그래머스] 2016년 basic

문제 :

programmers.co.kr/learn/courses/30/lessons/12901?

코딩테스트 연습 - 2016년

 

분석 :

윤년과 7월에서 8월로 넘어갈 시 홀수 짝수 계산을 달리 해준다. 

 

c++ 풀이 :

#include <string>
#include <vector>
#include <iostream>

using namespace std;

string solution(int a, int b) {
    string answer = "";
    int sum = 0;
    //int day = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
    if(a>1){
       for(int i =1; i < a; i++){
           if(i <= 7){
            if(i % 2 == 1) {
                sum += 31;
            } else{
                if (i == 2){
                    sum += 29;
                }else{
                    sum += 30;
                }
            } 
           }
           else{
              if(i % 2 == 0) {
                sum += 31;
            }else{
                sum += 30;
            }  
           }
       }
        
       sum += b;  
    } 
    else{
        sum = b;
    }
    
    if(sum % 7 == 1){
        answer = "FRI";
    }else if( sum % 7 == 2){
        answer = "SAT";
    }else if(sum % 7 == 3){
        answer = "SUN";
    }else if(sum % 7 == 4){
        answer = "MON";
    }else if(sum % 7 == 5){
        answer = "TUE";
    }else if(sum % 7 == 6){
        answer = "WED";
    }else if(sum % 7 == 0){
        answer = "THU";
    }
    
    return answer;
}