[백준] 주유소, boj_13305, 파이썬

https://www.acmicpc.net/problem/13305

13305번: 주유소

문제

어떤 나라에 N개의 도시가 있다. 이 도시들은 일직선 도로 위에 있다. 편의상 일직선을 수평 방향으로 두자. 제일 왼쪽의 도시에서 제일 오른쪽의 도시로 자동차를 이용하여 이동하려고 한다. 인접한 두 도시 사이의 도로들은 서로 길이가 다를 수 있다. 도로 길이의 단위는 km를 사용한다.

처음 출발할 때 자동차에는 기름이 없어서 주유소에서 기름을 넣고 출발하여야 한다. 기름통의 크기는 무제한이어서 얼마든지 많은 기름을 넣을 수 있다. 도로를 이용하여 이동할 때 1km마다 1리터의 기름을 사용한다. 각 도시에는 단 하나의 주유소가 있으며, 도시 마다 주유소의 리터당 가격은 다를 수 있다. 가격의 단위는 원을 사용한다.

예를 들어, 이 나라에 다음 그림처럼 4개의 도시가 있다고 하자. 원 안에 있는 숫자는 그 도시에 있는 주유소의 리터당 가격이다. 도로 위에 있는 숫자는 도로의 길이를 표시한 것이다. 

제일 왼쪽 도시에서 6리터의 기름을 넣고, 더 이상의 주유 없이 제일 오른쪽 도시까지 이동하면 총 비용은 30원이다. 만약 제일 왼쪽 도시에서 2리터의 기름을 넣고(2×5 = 10원) 다음 번 도시까지 이동한 후 3리터의 기름을 넣고(3×2 = 6원) 다음 도시에서 1리터의 기름을 넣어(1×4 = 4원) 제일 오른쪽 도시로 이동하면, 총 비용은 20원이다. 또 다른 방법으로 제일 왼쪽 도시에서 2리터의 기름을 넣고(2×5 = 10원) 다음 번 도시까지 이동한 후 4리터의 기름을 넣고(4×2 = 8원) 제일 오른쪽 도시까지 이동하면, 총 비용은 18원이다.

각 도시에 있는 주유소의 기름 가격과, 각 도시를 연결하는 도로의 길이를 입력으로 받아 제일 왼쪽 도시에서 제일 오른쪽 도시로 이동하는 최소의 비용을 계산하는 프로그램을 작성하시오.

입력

표준 입력으로 다음 정보가 주어진다. 첫 번째 줄에는 도시의 개수를 나타내는 정수 N(2 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 인접한 두 도시를 연결하는 도로의 길이가 제일 왼쪽 도로부터 N-1개의 자연수로 주어진다. 다음 줄에는 주유소의 리터당 가격이 제일 왼쪽 도시부터 순서대로 N개의 자연수로 주어진다. 제일 왼쪽 도시부터 제일 오른쪽 도시까지의 거리는 1이상 1,000,000,000 이하의 자연수이다. 리터당 가격은 1 이상 1,000,000,000 이하의 자연수이다. 

출력

표준 출력으로 제일 왼쪽 도시에서 제일 오른쪽 도시로 가는 최소 비용을 출력한다. 

 

 

문제풀이

처음 문제풀이를 할때, 주유비가 가장 적은 도시를 변수 X에 할당해 놓고 for 문을 돌면서 X가 나올때까지 해당되는 거리만 주유비를 더해준다. X가 나오면 남은 거리를 X를 곱해주고 마무리한다.

 

 

실패 코드

# 주유소
import sys
input = sys.stdin.readline

n = int(input())
km = list(map(int,input().split()))
city = list(map(int,input().split()))

# 마지막 도시는 상관없으므로
city.pop()

cost = 0
min_city = min(city) # 최소 주유비 도시를 min_city 변수에 할당
sum_km = sum(km)     # 거리의 총합
for i in range(n-1):
    if sum_km != 0:
        if city[i] == min_city:
            cost += sum_km * city[i]
            sum_km = 0
        else:
            cost += km[i] * city[i]
            sum_km -= km[i]
    else:
        break

print(cost)

 

하지만 이렇게 풀이하니, 서브테스크 2번에서 막히는 경험을 할 수 있었다. 서브테스크 2번이 수치가 높은것으로 보아 무언가 코드의 효율성면에서 떨어지는 것으로 판단되어 코드를 개선하고자했는데.. 개선이 안되서 결국 구글링..

 

최종 코드(참고풀이)

# 주유소
import sys
input = sys.stdin.readline

n = int(input())
km = list(map(int,input().split()))
city = list(map(int,input().split()))

pay = 0
cost = city[0]

for i in range(n-1):
    if city[i] < cost:
        cost = city[i]
    pay += cost * km[i]

print(pay)

이 코드를 살펴보니 최소로 주유를 할 수 있는 곳을 구해서 변수를 할당할 필요도, 남은 km 정보도 따로 저장하지 않아도 된다..엄청나구만..

 

이 풀이를 살펴보면 도시를 지나칠때마다 그 도시의 주유비를 cost 변수에 할당해놓고, 다음 도시의 주유비가 cost 보다 작다면 다시 cost 변수를 초기화 하면서 km를 곱해가면 된다. 이렇게 풀이하니.. 내가 스파게티 코드처럼 쫘놓은 것보다 훨씬 가독성도 좋고 간편하게 모든 과정을 수행할 수 있다....!