[swift] 백준 - 1874번: 스택 수열

1874번: 스택 수열

문제 푸는 데 있어 도움이 되도록 나의 풀이와 개선된 풀이를 올립니다.
또한 풀이 후 다른 사람의 풀이를 보고 참고할만한 풀이도 올립니다.
- 문제에 따라 나의 풀이만 있을 수 있습니다.
- 해당 풀이들은 풀이 중 하나일 뿐 최선의 풀이는 아닐 수 있습니다.

 

---

 

나의 풀이

import Foundation

let input = Int(readLine()!)!
var arr = (0..<input).map { _ in Int(readLine()!)! }
var stack = [Int]()
var result = [String]()
var count = 0
for i in 1...(input+1) { 
    guard !stack.isEmpty else {
        stack.append(i)
        result.append("+")
        continue
    }
    
    while count < arr.count && arr[count] == stack.last {
        stack.removeLast()
        count += 1
        result.append("-")
    }
    
    stack.append(i)
    result.append("+")
    
}
if stack.count == 1 {
    result.removeLast()
    for i in result {
        print(i)
    }
} else { print("NO") }

 

---

 

참고할만한 풀이

let n = Int(readLine()!)!
var stack:[Int] = [0]
var count = 0
var str = ""
for _ in 1...n {
    let num = Int(readLine()!)!
    while count < num {
        count += 1
        stack.append(count)
        str += "+\n"
    }
    if stack.popLast() != num {
        str = "NO"
        break
    }
    str += "-\n"
}
print(str)

 

- 저의 풀이와 알고리즘은 비슷하지만

훨씬 직관적이네요.

 

---