문제
총 N개의 시험장이 있고, 각각의 시험장마다 응시자들이 있다. i번 시험장에 있는 응시자의 수는 Ai명이다.
감독관은 총감독관과 부감독관으로 두 종류가 있다. 총감독관은 한 시험장에서 감시할 수 있는 응시자의 수가 B명이고, 부감독관은 한 시험장에서 감시할 수 있는 응시자의 수가 C명이다.
각각의 시험장에 총감독관은 오직 1명만 있어야 하고, 부감독관은 여러 명 있어도 된다.
각 시험장마다 응시생들을 모두 감시해야 한다. 이때, 필요한 감독관 수의 최솟값을 구하는 프로그램을 작성하시오.
입력
첫째 줄에 시험장의 개수 N(1 ≤ N ≤ 1,000,000)이 주어진다.
둘째 줄에는 각 시험장에 있는 응시자의 수 Ai (1 ≤ Ai ≤ 1,000,000)가 주어진다.
셋째 줄에는 B와 C가 주어진다. (1 ≤ B, C ≤ 1,000,000)
출력
각 시험장마다 응시생을 모두 감독하기 위해 필요한 감독관의 최소 수를 출력한다.
내 풀이
import sys
input = sys.stdin.readline
n = int(input())
students = list(map(int, input().split()))
b, c = map(int, input().split())
answer = 0
for s in students:
answer += 1
if s < b:
pass
else:
s -= b
if s%c == 0:
answer += s//c
else:
answer += (s//c + 1)
print(answer)코멘트
쉬운 문제지만 함정을 하나 조심해야 하는데 바로 한 고사장의 학생 수가 총감독관이 감독할 수 있는 학생 수보다 적은 경우이다. 이 때에는 총감독관만 있으면 되고, 부감독관은 굳이 없어도 되므로 총감독관의 수만 세도록 예외처리를 해줘야 한다.
s%c != 0 일때와 아닐 때를 나눌 필요 없이 간단하게 math.ceil로도 문제를 해결할 수 있다.