[PS] 99클럽 코테 스터디 28일차 TIL (괄호 회전하기)

문제

문제 링크

설명

입출력 예시 처럼 괄호 조합을 회전 시켰을 때 올바른 괄호 조합의 갯수를 반환해야 한다. 문자열의 길이만큼 맨 앞 괄호를 맨 뒤로 옮기는 작업을 반복 하면서 올바른 문자열인지 확인해야 한다.

풀이

올바른 괄호가 되기 위해선 여는 괄호('(', '{', '[') 와 닫는 괄호가 짝을 이뤄야 한다. 여는 괄호가 나올때마다 스택에 넣어뒀다가 닫는 괄호를 만나면 스택에 넣어두었던 여는 괄호를 꺼낸다. 스택이 모두 비었다면 문자열로 입력된 괄호가 모두 올바르다는 의미이므로 answer 를 증가시킨다. 이 작업을 문자열을 회전하면서 계속 반복한다.

class Solution {
    public int solution(String s) {
    int answer = 0;
    for (int i = 0; i < s.length(); i++) {
      String s1 = s.substring(i) + s.substring(0, i);
      if (check(s1)) {
        answer++;
      }
    }
    return answer;
  }

  boolean check(String s) {
    ArrayDeque<Character> stack = new ArrayDeque<>();
    char[] charArray = s.toCharArray();
    for (char ch : charArray) {
      if (ch == '(' || ch == '[' || ch == '{') {
        stack.push(ch);
      } else if (ch == ')') {
        if (stack.isEmpty() || !stack.pop().equals('(')) {
          return false;
        }
      } else if (ch == ']') {
        if (stack.isEmpty() || !stack.pop().equals('[')) {
          return false;
        }
      } else if (ch == '}') {
        if (stack.isEmpty() || !stack.pop().equals('{')) {
          return false;
        }
      }
    }
    return stack.isEmpty();
  }
}