[백준] 9012번: 괄호 - Kotlin[코틀린]

문제

 

9012번: 괄호

---

풀이

 

 이 문제는 스택을 이용하여 해결할 수 있다. '('을 입력받은 경우 스택에 저장하고 ')'을 입력받은 경우에는 스택에서 제거하는 방법이다. 이 때, 스택이 비어있을 때')'을 입력받은 경우와 모든 입력을 처리 했는데 스택이 비어있지 않는 경우에는 입력받은 괄호 문자열이 잘못된 문자열인 것이다.

 

 이번에는 mutableList를 이용해 스택을 구현해 보았다. 스택에서 push(n)에 해당하는 함수는 list.add(n)을 이용하고, pop()은 list.removeAt(list.size - 1)을 이용해 구현하면 mutableList로 스택을 구현할 수 있다.

 

---

코드

 

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()

    val testCase = br.readLine().toInt()

    val stack = mutableListOf<Int>() // stack 구현
    fun push(n: Int) = stack.add(n)
    fun pop(): Int {
        return if(stack.size==0) -1
        else {
            stack[stack.size - 1]
            stack.removeAt(stack.size-1)
        }
    }

    repeat(testCase) {
        for(ps in br.readLine()) {
            when(ps) {
                '(' -> push(1) // '(' 일 때, push(1)
                ')' -> { // ')' 일 때, 
                    if(stack.size==0) {  // stack이 비어있다면 잘못된 괄호 문자열이다.
                        push(1)
                        break
                    } else { // 아니라면 pop()
                        pop()
                    }
                }
            }
        }

        if(stack.size==0) {
            bw.append("YES\n")
        } else { // stack이 비어있지 않다면 NO를 출력하고 stack을 비워준다.
            bw.append("NO\n")
            stack.clear()
        }
    }

    bw.write("")
    bw.flush()
    bw.close()
    br.close()
}