[프로그래머스] 2016년

연습문제 - 2016년

문제 링크 : 2016년

사용 언어 : Python3

텍스트 슬라이싱을 쓰면 쉽다.

def solution(a, b):
    answer = ''
    days = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    
    count = sum(days[0:a-1]) + b - 1
    
    if count % 7 == 0 :
        answer += 'FRI'
    elif count % 7 == 1 :
        answer += 'SAT'
    elif count % 7 == 2 :
        answer += 'SUN'
    elif count % 7 == 3 :
        answer += 'MON'
    elif count % 7 == 4 :
        answer += 'TUE'
    elif count % 7 == 5 :
        answer += 'WED'
    else :
        answer += 'THU'
        
    return answer

사용 언어 : C++

#include <string>
#include <vector>

using namespace std;

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