[브론즈3/파이썬] 백준 2455 - 지능형 기차

 

문제

최근에 개발된 지능형 기차가 1번역(출발역)부터 4번역(종착역)까지 4개의 정차역이 있는 노선에서 운행되고 있다. 이 기차에는 타거나 내리는 사람 수를 자동으로 인식할 수 있는 장치가 있다. 이 장치를 이용하여 출발역에서 종착역까지 가는 도중 기차 안에 사람이 가장 많을 때의 사람 수를 계산하려고 한다. 단, 이 기차를 이용하는 사람들은 질서 의식이 투철하여, 역에서 기차에 탈 때, 내릴 사람이 모두 내린 후에 기차에 탄다고 가정한다.

 

 내린 사람 수탄 사람 수1번역(출발역)2번역3번역4번역(종착역)

0	32
3	13
28	25
39	0

예를 들어, 위와 같은 경우를 살펴보자. 이 경우, 기차 안에 사람이 가장 많은 때는 2번역에서 3명의 사람이 기차에서 내리고, 13명의 사람이 기차에 탔을 때로, 총 42명의 사람이 기차 안에 있다.

이 기차는 다음 조건을 만족하면서 운행된다고 가정한다.

1. 기차는 역 번호 순서대로 운행한다.
2. 출발역에서 내린 사람 수와 종착역에서 탄 사람 수는 0이다.
3. 각 역에서 현재 기차에 있는 사람보다 더 많은 사람이 내리는 경우는 없다.
4. 기차의 정원은 최대 10,000명이고, 정원을 초과하여 타는 경우는 없다.

4개의 역에 대해 기차에서 내린 사람 수와 탄 사람 수가 주어졌을 때, 기차에 사람이 가장 많을 때의 사람 수를 계산하는 프로그램을 작성하시오.

입력

각 역에서 내린 사람 수와 탄 사람 수가 빈칸을 사이에 두고 첫째 줄부터 넷째 줄까지 역 순서대로 한 줄에 하나씩 주어진다. 

출력

첫째 줄에 최대 사람 수를 출력한다.  

예제 입력 1 복사

0 32
3 13
28 25
39 0

예제 출력 1 복사

42

 

 

 

---

 

 

내 코드(36ms)

arr = []	# 증감 정도
brr = []	# 각 역의 인원수

for i in range(4):
    a, b = map(int, input().split())
    arr.append(-a+b)

brr.append(arr[0])
for i in range(1,4):
    brr.append(brr[i-1]+arr[i])

print(max(brr))

 

일일이 (n-1번째 역에서 출발할 때 남아있던 사람- n번째 역에서 내린 사람 + n번째 역에서 탄 사람)을 계산하기보다는, 각 역에 해당하는 증감으로 판단하면 된다.

==> (이해하기 어렵다면 문맥만) 등차수열의 등차 처럼 생각하면 될 거 같다! 물론 뜻은 다름

 

예제로 보면 이해하기 쉬운데

역 번호	1	2	3	4
내린 사람(-로 표시)	0	-3	-28	-39
탄 사람	32	13	25	0
남아있는 인원	32	42 = 32-3+13	39 = 42-28+25	0
증감		10 = 42-32	-3 = 39-42	-39 = 0-39

 

따라서 리스트 arr에 증감 정도(-a+b)를 넣어주었다.

그리고 각 역에 해당하는 인원수에 대한 brr이라는 리스트를 두었다.

 

이때 brr에 값을 넣는 for문을 설정하기 전, i=0일 때 brr[i-1]라는 계산이 불가한 동시에

brr[0] == arr[0]이므로 따로 빼주었다.

 

이후 2~4번역에 해당하는 인원수인 전 역(brr[i-1])+증감 정도(arr[i])라는 값을 brr에 넣어주면 된다.

 

마지막으로 max값으로 최대 인원수를 구한다.

 

 

 

 

다른 코드(32ms)

arr = []
n = 0

for _ in range(4):
    a, b = map(int, input().split())
    n += b
    n -= a
    arr.append(n)

print(max(arr))