선풍기를 몇대 사야 하나요
교실에 선풍기가 4대 있습니다. 선풍기는 한 대당 학생 k명에게 바람을 보냅니다. 모든 학생에게 바람을 보내기 위해서 선풍기를 몇 대 더 구매해야 할지 구하려고 합니다.
예를 들어, 선풍기 한 대당 학생 3명에게 바람을 보낼 수 있을 때, 한 교실당 학생 수가 [15, 17, 19, 10, 23] 명이고 선풍기가 4대 있으면 교실당 학생 [3, 5, 7, 0, 11] 명을 위해 선풍기를 더 구매해야 합니다. 이때, 교실별로 선풍기가 [1, 2, 3, 0, 4] 대씩 더 필요하니 총 선풍기 10대를 추가로 구매해야 합니다.
선풍기가 바람을 보내는 학생 수 k와 교실별 학생 수가 담긴 배열 student, 그리고 student의 길이인 student_len이 주어질 때 선풍기를 최소 몇 대 사야 하는지를 return 하도록 solution 함수를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다. 주어진 코드에서 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.
□ 매개변수 설명
선풍기가 바람을 보내는 학생 수 k와 교실별 학생 수가 담긴 배열 student, 그리고 student의 길이인 student_len이 solution 함수의 매개변수로 주어집니다.
선풍기가 바람을 보내는 학생 수 k는 1 이상 25 이하인 정수입니다.
한 교실엔 1명 이상 1,000명의 학생이 있습니다.
student_len은 100이하인 자연수입니다.
□ return 값 설명
선풍기를 최소 몇 대 사야 하는지를 return 합니다.
# -*- coding: utf-8 -*-
# UTF-8 encoding when using korean
def solution(k, student):
answer = 0
for s in student:
s -= 4*k
if s <= 0:
continue
answer += (s + k - 1) // k
return answer
