백준 4949 균형잡힌 세상(Java)

https://www.acmicpc.net/problem/4949

4949번: 균형잡힌 세상

---

 

해석 및 팁

 

---

 

이 문제에서 균형 잡힌 문자열인지 아닌지를 판단하기 위해문장을 입력받을때 괄호를 구별하여  ']' 의 경우는 '['과 매칭되어야 하고, ')'의 경우는 '('과 매칭되어야 균형을 이루고 있으므로 여는 경우에는 Stack에 저장해 주고 닫는 경우에 Stack이 비어있지 않거나 매칭되는 경우에는 여는 괄호를 지워주고 아닌 경우에는 닫는 괄호를 Stack에 저장하면 됩니다. 최종적으로 Stack에 값이 남아있으면 균형을 이루지 못하는 경우입니다.

---

 

Java 코드

 

---

import java.util.Scanner;
import java.util.Stack;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);

    while(true) {
      String str = sc.nextLine();
      
      if(str.equals(".")) break;
      
      Stack<Character> stack = new Stack<>();
      
      for(int i = 0; i < str.length(); i++) {
        char a = str.charAt(i);
        
        if(a == '(' || a == '[') stack.push(a);
        
        else if(a == ')') {
          if(!stack.isEmpty() && stack.peek() == '(') stack.pop();
          else stack.push(a);
        }
        
        else if(a == ']') {
          if(!stack.isEmpty() && stack.peek() == '[') stack.pop();
          else stack.push(a);
        }

      }
      if(stack.isEmpty()) System.out.println("yes");
      else System.out.println("no");
  
    }
  }
}