[swift] 백준 - 1138번: 한 줄로 서기

문제 링크

1138번: 한 줄로 서기

---

문제

N명의 사람들은 매일 아침 한 줄로 선다. 이 사람들은 자리를 마음대로 서지 못하고 오민식의 지시대로 선다.

어느 날 사람들은 오민식이 사람들이 줄 서는 위치를 기록해 놓는다는 것을 알았다. 그리고 아침에 자기가 기록해 놓은 것과 사람들이 줄을 선 위치가 맞는지 확인한다.

사람들은 자기보다 큰 사람이 왼쪽에 몇 명 있었는지만을 기억한다. N명의 사람이 있고, 사람들의 키는 1부터 N까지 모두 다르다.

각 사람들이 기억하는 정보가 주어질 때, 줄을 어떻게 서야 하는지 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 사람의 수 N이 주어진다. N은 10보다 작거나 같은 자연수이다. 둘째 줄에는 키가 1인 사람부터 차례대로 자기보다 키가 큰 사람이 왼쪽에 몇 명이 있었는지 주어진다. i번째 수는 0보다 크거나 같고, N-i보다 작거나 같다. i는 0부터 시작한다.

출력

첫째 줄에 줄을 선 순서대로 키를 출력한다.

예제

---

성공 풀이

let N = Int(readLine()!)!
let arr = readLine()!.split { $0 == " " }.map { Int($0)! }
var result = Array(repeating: 0, count: N)

loop: for i in 0..<N {
    var count = 0
    
    for j in 0..<N {
        if count == arr[i] {
            if result[j] == 0 {
                result[j] = i+1
                break
            }
            continue
        }
        if result[j] == 0 { count += 1 }
    }
}
result.forEach { print($0,terminator: " ")}

풀이 키워드

• N만큼 0으로 채워진 배열 result를 만들어 줬습니다. 입력으로 주어진 배열 arr을 순차적으로 탐색합니다. 배열의 값은 result에 들어가기 위한 값이 되는데 주어진 숫자가 2라면 result에서 0 두 개 뒤에 숫자를 넣어주게 됩니다. 예시를 들어보겠습니다.
• 예제가 2 1 1 0 일 때 result는 [0,0,0,0]으로 초기화가 됩니다.
• 예제 2 1 1 0 을 순차 적으로 돌 때 result가 변화는 모습을 보게 되면, 2일 때 result배열의 0이 2 개수 뒤에 숫자가 입력되기에 [0,0,1,0] 이 됩니다. 이어서 1 일 때는 2 앞에 본인의 숫자보다 큰 숫자가 한 개 밖에 안 오는 것이기 때문에 [0,2,1,0]이 됩니다. 이어서 1일 때는 [0,2,1,3] , [4,2,1,3]으로 result가 채워지게 되어 올바른 출력값을 도출할 수 있습니다.