백준 - 한 줄로 서기(Swift)

백준 - 한 줄로 서기(Swift)

 

 

 문제 설명

 

https://www.acmicpc.net/problem/1138

 

 나의 풀이

 

문제에서 주어진 수들을 이미 정렬된 상태의 수들이다.

즉 1번~N번까지 키가 순서대로 있는 것이다.

 

그 상태에서 자신의 왼쪽에 몇명이 있는지를 구하는 문제인데, 이는 공백을 체크하면서 해결할 수 있다.

예를들어 2 1 1 0의 순으로 입력이 들어왔을때,

첫번째 사람은 앞에 두명이 있다. 이는 _ _ 1 _ 의 형태가 되는 것이다. 

두번째 사람은 앞에 한명이 있기에 _ 2 1 _ 가되고,

세번째 사람은 앞에 한명이 있기 때문에 _ 2 1 3이 된다.

즉 들어오는 숫자는 앞의 공백의 개수라고 생각할 수 있다. 

 

이를 코드로 풀어내면 공백을 마주칠때 더 이상 앞에 더 큰 숫자가 있으면 안된다면 값을 지정하고, 공백을 마주칠때 앞에 더 큰 숫자가 있어야한다면 카운트만 줄여주는 것이다. 그리고 이미 지정된 값을 만나면 뒤에 입력받는 값들은 최소한 이 값보다는 크기 때문에 cnt를 건드리지않고 넘겨주기만 하면 된다.

 

 

import Foundation

let n = Int(readLine()!)!
let arr = readLine()!.components(separatedBy:" ").map{Int($0)!}
var visited = [Int](repeating: 0, count: n)

for (i, x) in arr.enumerated() {
    var cnt = x
    for j in 0..<arr.count {
        if visited[j] == 0 && cnt == 0 { // 공백이면서 더 이상 앞에 큰 것이 없을때 고정
            visited[j] = i+1 
            break
        } else if visited[j] == 0 { // 공백이지만 앞에 더 큰 숫자가 있을때 더 큰 숫자가 몇개인지를 알려주는 변수의 수만 줄여준다.
            cnt -= 1
        } else { // 숫자가 이미 배정되어있을때 지나친다.
            continue 
        }
    }
}

print(visited.map{String($0)}.joined(separator:" "))