[BOJ23971/Python] ZOAC 4

1. 문제 링크 : https://www.acmicpc.net/problem/23971

23971번: ZOAC 4

 

2. W개씩 H행에 걸쳐 있을 때, 모든 참가자는 세로로 N칸 또는 가로로 M칸 이상 비우고 앉아야 한다.

 

3. 메모리 초과 코드

# col : W , row : H
# 세로 N or 가로 M 띄고 앉기
input_lst = list(map(int, input().split()))
H, W, N, M = input_lst[0], input_lst[1], input_lst[2], input_lst[3] # 5 4 1 1
seats = [[0]*W for _ in range(H)]
# 한 명이 자리에 앉는 경우 주위를 -1 로 만들어서 앉을 수 없도록 해주기
for r in range(0, H, M+1):
    for c in range(0, W, N+1):
        seats[r][c] = 1
cnt = 0
for r in range(H):
    for c in range(W):
        if seats[r][c] == 1:
            cnt += 1
print(cnt)

 

4. 정답 코드

import math
H, W, N, M = map(int, input().split())
a = math.ceil(W/(M+1))
b = math.ceil(H/(N+1))
print(a*b)

 

5. 고찰

메모리 제한 : 512MB

시간 제한 : 1초