[99클럽] 15일차 문제: 균형잡힌 세상

 

백준 균형잡힌 세상은 ()[] 괄호가 나오는 스택 자료구조를 사용하는 기본적인 문제였다. 중요 로직은 isBalanced 부분인데 열린 괄로면 스택에 넣고 닫힌 가로이면 스택에서 pop을 한 다음 쌍이 맞는지 확인하면 된다. 이때 스택이 비어있는지도 잘 확인는 로직도 잘 작성해주어야 한다. 

 

아래 코드는 지피티가 열심히 작성해준 답이다

import java.io.*;
import java.util.*;

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) {
            String line = br.readLine();
            if (line.equals(".")) break;
            sb.append(isBalanced(line) ? "yes" : "no").append("\n");
        }
        System.out.print(sb);
    }
    
    private static boolean isBalanced(String str) {
        Stack<Character> stack = new Stack<>();
        
        for (char ch : str.toCharArray()) {
            if (ch == '(' || ch == '[') {
                stack.push(ch);
            } else if (ch == ')') {
                if (stack.isEmpty() || stack.pop() != '(') {
                    return false;
                }
            } else if (ch == ']') {
                if (stack.isEmpty() || stack.pop() != '[') {
                    return false;
                }
            }
        }
        
        return stack.isEmpty();
    }
}