[백준] 4949번 : 균형잡힌 세상

4949번: 균형잡힌 세상

---

문제

---

예제

---

풀이

9012번 괄호 문제의 업그레이드 버전인 문제라고 할 수 있다.

[백준] 9012번 : 괄호

문자열 입력을 받고 해당 문자열이 .이면 종료하고 버퍼에 있는 걸 출력한다.

여는 괄호와 여는 대괄호는 push()를 통해 스택에 넣어주고,

닫는 괄호를 입력받으면 스택이 비어있거나, pop()할 원소가 여는 괄호인지 peek()를 통해서 확인한다.

대괄호 또한 위와 같이 확인하고, 만약 위와 같은 상황일시 "no"를 return 한다.

마지막으로 조건문을 통해 스택이 비어있으면 "yes", 남아있으면 "no"를 반환한다.

---

소스 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        while (true) {
            Stack<Character> stack = new Stack<>();
            String str = br.readLine();

            if (str.equals(".")) break;

            sb.append(bracket(str)).append('\n');
        }
        System.out.println(sb);
    }
    public static String bracket(String str) {
        Stack<Character> stack = new Stack<>();

        for (int i = 0; i < str.length(); i++) {
            char c = str.charAt(i);

            if (c == '(' || c == '[') {
                stack.push(c);
            }
            else if (c == ')') {
                if (stack.empty() || stack.peek() != '(') {
                    return "no";
                }
                else stack.pop();
            }
            else if (c == ']') {
                if (stack.empty() || stack.peek() != '[') {
                    return "no";
                }
                else stack.pop();
            }
        }

        if (stack.empty()) return "yes";
        else return "no";
    }
}