[백준] 지능형 기차 2 (Python)

• https://www.acmicpc.net/problem/2460

2460번: 지능형 기차 2

문제

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

내린 사람 수 탄 사람 수

1번역(출발역)	0	32
2번역	3	13
3번역	28	25
4번역	17	5
5번역	21	20
6번역	11	0
7번역	12	12
8번역	4	2
9번역	0	8
10번역(종착역)	21	0

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

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

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

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

입력

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

출력

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

입출력 예제

입력	출력
0 32 3 13 28 25 17 5 21 20 11 0 12 12 4 2 0 8 21 0	42

풀이

각 역마다 타고 있는 사람 수를 계산하여 배열로 저장한 후 최댓값을 출력한다.

각 역의 타고 있는 사람 수는 직전 역에서의 사람 수 - 내린 사람 수 + 탄 사람 수다.

import sys
input = sys.stdin.readline
info = [list(map(int, input().split())) for _ in range(10)]
nums = [info[0][1]] * 10 # 첫번째 역에서 탄 사람 수로 초기화 
for i in range(1, 10):
	nums[i] = nums[i-1] - info[i][0] + info[i][1]
print(max(nums))

시간 복잡도: O(n)

공간 복잡도: O(n)

메모리: 31256 KB / 시간: 40ms

 

max_num이라는 변수를 선언하여 현재까지의 max값을 저장하면 O(1)의 공간복잡도로 풀 수 있겠다는 생각이 들었다.

import sys
input = sys.stdin.readline
info = [list(map(int, input().split())) for _ in range(10)]
max_num = temp = 0
for i in info:
	temp += i[1] - i[0]
	max_num = max(max_num, temp)
print(max_num)

시간 복잡도: O(n)

공간 복잡도: O(1)

메모리: 31256 KB / 시간: 40ms

 

표기나 초기화가 좀 더 파이썬스러운 코드라고 생각한다. 포인트는 이전역에서 탄 사람의 수가 temp에 저장되어 있어 연산이 유기적이라는 점!