plzy의 개발 블로그

[백준 실버 4] 괄호 - kotlin 본문

알고리즘/coding test

[백준 실버 4] 괄호 - kotlin

plzyhappy 2022. 4. 19. 10:03

[Silver IV] 괄호 - 9012

문제 링크

성능 요약

메모리: 12484 KB, 시간: 88 ms

분류

자료 구조(data_structures), 스택(stack), 문자열(string)

문제 설명

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다.

여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나타내어야 한다.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 주어진다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 괄호 문자열이 한 줄에 주어진다. 하나의 괄호 문자열의 길이는 2 이상 50 이하이다.

출력

출력은 표준 출력을 사용한다. 만일 입력 괄호 문자열이 올바른 괄호 문자열(VPS)이면 “YES”, 아니면 “NO”를 한 줄에 하나씩 차례대로 출력해야 한다.

풀이

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter
import java.lang.Exception
import java.util.*

fun main() {

    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))
    val input = br.readLine().toInt()
    repeat(input) {
        val ps = br.readLine().toString()
        val psStack = Stack<Char>()
        try {
            for (i in ps) {

                when (i) {
                    '(' -> {
                        psStack.push(i)
                    }
                    ')' -> {
                        psStack.pop()
                    }
                }
            }
            if (psStack.isEmpty()) {
                bw.write("${"YES"}\n")
            } else {
                bw.write("${"NO"}\n")
            }

        } catch (e: Exception) {
            bw.write("${"NO"}\n")


        }
    }
    bw.flush()

}

stack을 이용해 '(' 일때는 push하고 ')' 일땐 pop을 해 사이즈를 비교했다.
비어있다면 짝기맞는거니 YES, 비어있지않다면 짝이 맞지 않으므로 NO다.
또한 pop할때 예외처리를 안하면 EmptyStackException 오류가 뜨므로 예외처리를 해줬다.