정올 jungol 2078 13일의 금요일 Java, Python

문제로이동

JUNGOL

---

목차

 

 

---

문제 설명

모듈러 연산을 하는 문제입니다.

 

N의 범위가 명시되지 않아 TLE를 예측할 수 없지만, 혹시 모를 효율성을 위해 모듈러 연산을 해주었습니다.

자세한 설명은 소스코드에 주석을 달아두었습니다.

---

소스 코드

Python

N = int(input())
# answer : 월,화,수,목,금,토,일의 횟수 저장
answer = [0]*7

# 12개월 초기 세팅
# month[0]은 0 처리
# 구현의 편의성을 위하여 1월~12월의 인덱스를 갖는다. (0~12)
# 예시) 1월은 index 0이 아니라 1을 가리킨다. 12월은 index 12를 가리킨다.
month = [31]*13
month[0]=0
month[4]=30 # 4,6,9,11월은 30일
month[6]=30
month[9]=30
month[11]=30

# day : 요일을 가리킨다 (월,화,수,목,금,토,일)
# 월요일은 0이다.
# 모듈러 연산을 통해 요일을 골라낼 수 있다.
day = 0 

for year in range(1900,1900+N):
    month[2] = 28   #평년은 2월이 28일
    if (year%4==0 and year%100!=0) or year%400==0:  #윤년의 조건(문제에 명시되어 있음)
        month[2] = 29   #윤년은 2월이 29일
    for mth in range(1,13):
        # -1을 하는 이유는 
        # 1900년 1월 1일에서 13번째 요일을 계산해보면 쉽게 알 수 있다.
        # -1을 하지 않으면 6으로 일요일을 가리키게 되므로 13번째 요일인 토요일을 가리키려면 -1을 해줘야 한다.
        answer[(day+13)%7-1]+=1     

        day = (day+month[mth])%7    #1개월 후의 시작 요일로 갱신해 준다.

for i in answer:    #횟수 계산이 끝났다면 최종적으로 포맷에 맞게 출력한다.
    print(i,end=' ')

---

Java8

import java.util.Arrays;
import java.util.Scanner;

public class Main_jungol_2078_13일의금요일 {	
	public static void main(String[] args) {		
		Scanner scan = new Scanner(System.in);
		int N = scan.nextInt();
		int[] month = new int[13];	//1월~12월까지 일 수를 표기할 배열 생성
		// 1월은 31일, 2월은 28일 또는 29일 ,,,
		Arrays.fill(month, 31);
		month[4]=30;//4,6,9,11월은 30일(문제에 명시)
		month[6]=30;
		month[9]=30;
		month[11]=30;
		
		int answer[] = new int[7];//각 요일의 횟수를 저장(index는 day변수와 동일)
		int day = 0;//요일을 저장 (월요일:0, 일요일:6)
		
		//1900년 ~ 1900+N-1년까지 계산
		for(int year=1900;year<1900+N;year++) {
			month[2] = 28;//평년은 2월이 28일
			if((year%4==0 && year%100!=0)||year%400==0)//윤년인 경우
				month[2] = 29;//윤년은 2월이 29일
			
			for(int mth=1;mth<=12;mth++) {//12개월 순환
// -1을 하는 이유는 
// 1900년 1월 1일에서 13번째 요일을 계산해보면 쉽게 알 수 있다.
// -1을 하지 않으면 6으로 일요일을 가리키게 되므로 13번째 요일인 토요일을 가리키려면 -1을 해줘야 한다.
// java의 경우 python과 달리 음수가 되는 경우도 생각해야 한다. 예) (day+13)%7이 0이 되는 경우
				answer[(day+13)%7==0?6:(day+13)%7-1]++;				
				day = (day+month[mth])%7;//1개월 후의 `시작 요일`로 갱신해 준다.
			}
		}
		
		for(int count:answer) {
			System.out.print(count+" ");
		}
	}
}