[알고리즘] 백준 10810번 : 공 넣기

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.

공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀져 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)

도현이는 입력으로 주어진 순서대로 공을 넣는다.

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.

예제 입력 1

5 4
1 2 3
3 4 4
1 4 1
2 2 2

예제 출력 1

1 2 1 1 0

 

풀이

import Foundation

let input = readLine()!.split(separator: " ").map { Int($0)! }
let N = input[0]
let M = input[1]

// array 초기화
var array: [Int] = []
for _ in 0...N - 1 {
    array.append(0)
}

//ijk방식으로 공 넣기
for _ in 0..<M {
    let ijk = readLine()!.split(separator: " ").map { Int($0)! }
    for index in ijk[0]...ijk[1] {
        array[index - 1] = ijk[2]
    }
}

array.forEach { ballNum in
    print(ballNum, terminator: " ")
}

1. array에 0을 N개 채움니다

2. i 부터 j까지 k로 대체합니다.

 

• array를 for문을 사용하는게 아닌 더 효율 적인 방법으로 초기화 할 수 있을 것 같아서 찾아봤더니 역시나 있었다!

var array = [Int](repeating: 0, count: N)

이렇게 하면 array를 초기화할 때 for문을 ㅅ용하지 않아도 된다.

replaceSubrange를 사용하는 방법도 찾았다.

코드상에서는 이해하기 더 복잡하고 개인적으로는 직관적이지 않다고 생각했다.

메모리나 시간적으로 이점이 크지 않다면 굳이 쓰고 싶지 않았지만..

채점 시 메모리나 시간적으로 훨씬 이점이 있었다.

만도스님의 블로그에서 그 답을 확인할 수 있었다.

 

끝!