[백준] 9012. 괄호 (Kotlin)

문제

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

9012번: 괄호

 

풀이

0. 문제 해석

괄호는 짝이 맞아야 하므로 스택을 활용해야 한다.

 

1. 여는 괄호면 스택에 추가한다.
2. 닫는 괄호일 때, 스택이 비어있거나 스택에서 pop 하여 그 값이 닫는 괄호면  NO 리턴
3. 전체 괄호 검사 후 스택에 남아있는 문자가 있을 경우 NO 리턴
4. 2,3번에서 리턴되지 않았을 경우 YES 리턴

 

 

1. 전체 코드

// [백준] 9012. 괄호 (Kotlin)
import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.*

fun main() = with(BufferedReader(InputStreamReader(System.`in`))){
    val n = readLine().toInt()

    for(i in 0 until n) {
        val str = readLine()
        solve(str)
    }
}

fun solve(str: String){
    val stack = Stack<Char>()

    for(i in str){
        // 여는 괄호면 추가
        if(i == '('){
            stack.add(i)
        }
        // 닫는 괄호일 때
        else if(i == ')'){
            // 비어 있을 때
            if(stack.isEmpty()){
                println("NO")
                return
            }
            // 짝이 맞지 않을 때
            if(stack.pop() == ')'){
                println("NO")
                return
            }
        }
    }

    // 연산 종료 시 스택이 비어있어야 YES
    if(stack.isEmpty()){
        println("YES")
    }
    else{
        println("NO")
    }
}