프로그래머스 Lev.1 2016년 [C언어]

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


char* solution(int a, int b) {
    char* answer = (char*)malloc(sizeof(char)*3);
    char* day[] = { "MON","TUE","WED","THU","FRI","SAT","SUN"};

    
    int k = 4;  //금요일 friday
    bool isDay = false;
    for (int i = 1; i <= a; i++) 
    {
        if (i == 1 || i == 3 || i == 5 || i == 7 || i == 8 || i == 10 || i == 12)
        {
            for (int j = 1; j <= 31; j++) 
            {
                if (i == a && b == j)
                {
                    isDay = true; break;
                }
                k++;
                if (k == 7) k = 0;
            }
        }
        else if (i == 2) 
        {
            for (int j = 1; j <= 29; j++)
            {
                if (i == a && b == j)
                {
                    isDay = true; break;
                }
                k++;
                if (k == 7) k = 0;
            }
        }
        else 
        {
            for (int j = 1; j <= 30; j++)
            {
                if (i == a && b == j)
                {
                    isDay = true; break;
                }
                k++;
                if (k == 7) k = 0;
            }
        }
        if (isDay == true) break;
    }

    answer = day[k];

    return answer;
}

2023-03-27