21.06.24 - [PG] 2016년

[suhyunsim]

문제

• 2016년

핵심 아이디어

어려운 점, 실수

• 1월 1일 count를 고려하지 못했다.(1월 1일은 첫번재 요일이 되어야 하는데 카운트 되어버림 -> -1해서 고려해줘야 함)

풀이

package main.java.com.poogle.PG.Q12901;

public class Solution {
    public static void main(String[] args) {
        System.out.println(solution(5, 24));
    }

    public static String solution(int a, int b) {
        String[] year = new String[366];
        int[] month = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
        String[] days = {"SUN", "MON", "TUE", "WED", "THU", "FRI", "SAT"};
        for (int i = 0; i < year.length; i++) {
            year[i] = days[(i + 5) % 7];
        }
        int ans = 0;
        for (int i = 0; i < a - 1; i++) {
            ans += month[i];
        }
        ans += b - 1;
        return year[ans];
    }
}