[Swift] 5. 백준 1874번: 스택 수열

 

문제 링크

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

1874번: 스택 수열

 

---

풀이

 

 문제를 이해하는 것부터 꽤 시간이 걸렸다.

 push를 할수록 count의 값을 증가시켜서 입력 받은 숫자가 count보다 작을 때에는 계속 push를 해주고,

 입력받은 숫자가 stack의 마지막과 같을 때에는 pop을 시켜주었다.

 

 

 

 

---

 

전체 코드

 

import Foundation

let N = Int(readLine()!)!
var value : [String] = []
var stack : [Int] = []
var count = 1

for _ in 0...N-1 {
  let num = Int(readLine()!)!
  
  while count <= num {
    stack.append(count)
    value.append("+")
    count+=1
  }
  if stack.last == num {
    stack.popLast()
    value.append("-")
  }else {
    print("NO")
    exit(0)
  }
}

print(value.joined(separator: "\n"))

 

 

 

---