[프로그래머스] 2016년

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

코딩테스트 연습 - 2016년

 

#include <stdio.h>
#include <stdlib.h>


char* solution(int a, int b) {
    int month[13] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12 };
    int days[13] = { 0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
    char *DoW[] = { "SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT" };
    int the_day = 5;

    

    for (int M = 1; M <= a; M++)
    {
        if (M == a)
        {
            for (int D = 1; D < b; D++) // a달까지 돌면 b일까지만 돈다
            {
                if (the_day == 6)
                    the_day = -1;
                the_day = the_day + 1;
                //printf("%d %d \n", M, D);
            }
            

        }
        else
        {
            for (int D = 1; D <= days[M]; D++) // n월의 n일까지만 돈다
            {
                if (the_day == 6)
                    the_day = -1;
                the_day = the_day + 1;
                //printf("%d %d \n", M, D);
            }
            
        }
    }

    
    //printf("%s", DoW[the_day]);
    // 리턴할 값은 메모리를 동적 할당해주세요.
  
    char* answer = (char*)malloc(3);
    answer = DoW[the_day];
    return answer;
}


int main(void)
{
    printf("%s", solution(5, 24));
    return 0;
}