프로그래머스 괄호 회전하기 kotlin (스택) 2022-06-24 코드 추가

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/76502

코딩테스트 연습 - 괄호 회전하기

문제 설명

다음 규칙을 지키는 문자열을 올바른 괄호 문자열이라고 정의합니다.

• (), [], {} 는 모두 올바른 괄호 문자열입니다.
• 만약 A가 올바른 괄호 문자열이라면, (A), [A], {A} 도 올바른 괄호 문자열입니다. 예를 들어, [] 가 올바른 괄호 문자열이므로, ([]) 도 올바른 괄호 문자열입니다.
• 만약 A, B가 올바른 괄호 문자열이라면, AB 도 올바른 괄호 문자열입니다. 예를 들어, {} 와 ([]) 가 올바른 괄호 문자열이므로, {}([]) 도 올바른 괄호 문자열입니다.

대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• s의 길이는 1 이상 1,000 이하입니다.

---

입출력 예

---

입출력 예 설명

입출력 예 #1

• 다음 표는 "[](){}" 를 회전시킨 모습을 나타낸 것입니다.

• 올바른 괄호 문자열이 되는 x가 3개이므로, 3을 return 해야 합니다.

입출력 예 #2

• 다음 표는 "}]()[{" 를 회전시킨 모습을 나타낸 것입니다.

• 올바른 괄호 문자열이 되는 x가 2개이므로, 2를 return 해야 합니다.

입출력 예 #3

• s를 어떻게 회전하더라도 올바른 괄호 문자열을 만들 수 없으므로, 0을 return 해야 합니다.

입출력 예 #4

• s를 어떻게 회전하더라도 올바른 괄호 문자열을 만들 수 없으므로, 0을 return 해야 합니다.

---

※ 공지 - 2021년 4월 16일 테스트케이스가 추가되었습니다.

 

풀이

'프로그래머스 / 월간 코드 챌린지 시즌2 / 괄호 회전하기'로 분류되어 있는 문제이다.

주어진 문자열 s를 x(s.length-1)만큼 회전시키며, 해당 문자열이 올바른 괄호 문자열인지 확인하면 된다.

x만큼 회전한 문자열이 올바른 괄호 문자열인지 확인하는 방법은 스택을 이용했다.

ch가 [ ( { 처럼 여는 괄호면 stack에 push하고, stack이 비어있을 때는 ch가 무엇인지에 상관없이 ch를 stack에 push한다.

만약 stack이 비어있지 않고, stack의 윗부분과 ch가 짝이 맞는다면 (ex (==), {==}, [==] ) stack을 pop한다.

s[0]부터 s[length-1]까지 검사하여 스택이 비어있다면 answer를 1증가시킨다.

 

 

 

코드1

import java.util.Stack

class Solution {
    fun solution(s: String): Int {
        var answer: Int = 0
        var temp = s
        for(i in 0 until s.length){
               var stk = Stack<Char>()
            for(ch in temp){
                if(ch=='['||ch=='('||ch=='{'){
                    stk.push(ch)
                }
                else if(stk.empty()){
                    stk.push(ch)
                }
                else{
                    if((stk.peek()=='['&&ch==']')||(stk.peek()=='('&&ch==')')||(stk.peek()=='{'&&ch=='}')){
                        stk.pop()
                    }
                }
            }
            if(stk.empty()){
                answer++
            }
            temp = temp[temp.length-1]+temp.substring(0,temp.length-1)
        }

        return answer
    }
}

코드2

import java.util.*
class Solution {
    fun solution(s: String): Int {
        val sb = StringBuilder(s)
        var answer = 0
        label@for(i in s.indices){
            //rotate
            sb.append(sb[0])
            sb.deleteAt(0)
            //check
            val stk = Stack<Char>()
            for(ch in sb){
                //스택이 빈 경우
                if(stk.isEmpty()){
                    when(ch){
                        ']' -> continue@label
                        ')' -> continue@label
                        '}' -> continue@label
                        else -> stk.add(ch)
                    }
                }
                //스택이 비지 않은 경우
                else{
                    val bef = stk.peek()
                    when {
                        bef == '(' && ch == ')' -> stk.pop()
                        bef == '[' && ch == ']' -> stk.pop()
                        bef == '{' && ch == '}' -> stk.pop()
                        ch =='(' || ch == '[' || ch == '{' -> stk.add(ch)
                        else -> continue@label
                    }
                }
            }
            if(stk.isEmpty()) answer++
        }
        
        return answer
    }
}