[백준 / 구현] 2979 : 트럭 주차 (python)

난이도 : 브론즈 2

 

문제 설명

문제

상근이는 트럭을 총 세 대 가지고 있다. 오늘은 트럭을 주차하는데 비용이 얼마나 필요한지 알아보려고 한다.

상근이가 이용하는 주차장은 주차하는 트럭의 수에 따라서 주차 요금을 할인해 준다.

트럭을 한 대 주차할 때는 1분에 한 대당 A원을 내야 한다. 두 대를 주차할 때는 1분에 한 대당 B원, 세 대를 주차할 때는 1분에 한 대당 C원을 내야 한다.

A, B, C가 주어지고, 상근이의 트럭이 주차장에 주차된 시간이 주어졌을 때, 주차 요금으로 얼마를 내야 하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 문제에서 설명한 주차 요금 A, B, C가 주어진다. (1 ≤ C ≤ B ≤ A ≤ 100)

다음 세 개 줄에는 두 정수가 주어진다. 이 정수는 상근이가 가지고 있는 트럭이 주차장에 도착한 시간과 주차장에서 떠난 시간이다. 도착한 시간은 항상 떠난 시간보다 앞선다. 입력으로 주어지는 시간은 1과 100사이 이다.

출력

첫째 줄에 상근이가 내야하는 주차 요금을 출력한다.

예제 입력 1

5 3 1
1 6
3 5
2 8

예제 출력 1

33

예제 입력 2

10 8 6
15 30
25 50
70 80

예제 출력 2

480

 

풀이

A, B, C = map(int, input().split())
time = [0 for _ in range(100)]

for i in range(3):
    start, end = map(int, input().split())
    for j in range(start - 1, end - 1):
        time[j] += 1

price = 0

for truck in time:
    if truck == 1:
        price += A
    elif truck == 2:
        price += B*2
    elif truck == 3:
        price += C*3

print(price)

 

시간은 1에서 100으로 한정되어 있어서 인덱스를 100으로 가지는 배열 time을 선언했다.

반복문을 돌면서 각 트럭의 도착 시간과 떠난 시간을 start, end에 입력받는다.

start부터 end까지 반복문을 돌면서 해당 시간에 트럭이 몇 대 있는지 time 배열에 저장한다.

 

각 시간별 저장된 트럭 수에 따라 주차 요금이 달라지므로 이를 조건문으로 분류하여 가격을 계산한다.

price에 가격(A, B, C 중 하나) * 트럭수 를 더해 계산하면 된다.