[COS PRO 2급 #5-7] 계산시간을 구해주세요!

문제설명

XX 마트에선 구매할 물건이 3개 이하이면 소량 계산대에서, 그렇지 않으면 일반 계산대에서 계산해야 합니다. 두 계산대 모두 물건 한 개를 계산하는 데 1분이 걸립니다.

손님들이 구매할 물건 수가 담긴 리스트 stuffs가 매개변수로 주어질 때, 모든 물건을 계산하는 데 필요한 시간을 return 하도록 solution 함수를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다. 주어진 코드에서 _**한 줄**_만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.

매개변수 설명

손님들이 구매할 물건 수가 담긴 리스트 stuffs가 solution 함수의 매개변수로 주어집니다.
* 계산을 할 손님은 1명 이상 100명 이하입니다.
* 손님이 구입할 물건은 1개 이상 20개 이하입니다.

return 값 설명

모든 물건을 계산하는데 걸리는 시간을 return 해주세요.

예시

| stuffs          | return |
|--------------------|--------|
| [5, 3, 4, 2, 3, 2] | 10  |

예시설명

첫 번째 손님, 세 번째 손님은 일반 계산대에서 계산합니다. 이때 9분이 걸립니다.
두 번째, 네 번째, 다섯 번째, 여섯 번째 손님은 소량 계산대에서 계산합니다. 이때 10분이 걸립니다.
따라서 모든 물건을 계산하는데 걸리는 시간은 10분입니다.

문제 코드

def solution(stuffs):
    answer = 0
    small_counter, general_counter = 0, 0
    for s in stuffs:
        if s > 3:
            general_counter += s
        else:
            small_counter += s
    if small_counter < general_counter:
        answer = small_counter
    else:
        answer = general_counter
    return answer

풀이

def solution(stuffs):
    answer = 0
    small_counter, general_counter = 0, 0
    
    # 물건의 개수가 3개 이하면 small_counter, 그렇지 않으면 general_counter
    for s in stuffs:
        if s > 3:
            general_counter += s
        else:
            small_counter += s
            
    # 계산에 걸리는 시간은 small_counter, general_counter 중 큰 수
    if small_counter > general_counter:
        answer = small_counter
    else:
        answer = general_counter
    return answer