본문 바로가기
알고리즘

[Swift] 백준 2460번 지능형 기차 2

by 고고 2022. 3. 18.

링크: https://www.acmicpc.net/problem/2460

 

2460번: 지능형 기차 2

최근에 개발된 지능형 기차가 1번역(출발역)부터 10번역(종착역)까지 10개의 정차역이 있는 노선에서 운행되고 있다. 이 기차에는 타거나 내리는 사람 수를 자동으로 인식할 수 있는 장치가 있다.

www.acmicpc.net

 

 

내린 사람 수만큼 빼고, 탄 사람 수만큼 더해서 구했습니다.

가장 큰 값은 result에 저장했다가 출력합니다.

import Foundation

var trains = [[Int]]()
var people = [Int]()

for _ in 0..<10 {
    trains.append(readLine()!.split(separator: " ").map { Int(String($0))! })
}

var current = 0
var result = 0

for train in trains {
    current -= train[0] // 내린 사람 수
    current += train[1] // 탄 사람 수
    
    people.append(current)
    result = max(result, current)
}

print(result)

댓글