[백준 4949] 균형잡힌 세상

문제

 

4949번: 균형잡힌 세상

접근

 

 여는 괄호 ( [ 가 나오면 push를 한다.

 닫힌 괄호 ) ]가 나오면 pop을 한다.

    - 이때 pop을 하기전에 스택이 비어있으면 유효하지 않다. 

    - pop을 한 괄호의 짝이 안맞으면 유효하지 않다.

 

 모든 문자열에 대한 push, pop이 끝났는데 stack이 비어있지 않으면 유효하지 않다. 

 

 

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

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		String line = null;
		while (!(line = br.readLine()).equals(".")) {
			String result = getResult(line);
			bw.write(result+"\n");
			bw.flush();
		}

	}

	public static String getResult(String line) {
		String result = null;
		Stack<Character> stack = new Stack<Character>();
		for (int i = 0; i < line.length(); i++) {
			char c = line.charAt(i);
			
			if (c == '(') {
				stack.push('(');
			} 
			else if (c == '[') {
				stack.push('[');
			} 
			else if (c == ')') {
				if (stack.empty()) {
					result = "no";
					return result;
				}
				char popChar = stack.pop();
				if (popChar != '(') {
					result = "no";
					return result;
				}
			} 
			else if (c == ']') {
				if (stack.empty()) {
					result = "no";
					return result;
				}
				char popChar = stack.pop();
				
				if (popChar != '[') {
					result = "no";
					return result;
				}
			} else {
				continue;
			}
		}
		if (!stack.empty()) {
			result = "no";
		}
		else {
			result = "yes";
		}
		return result;
	}
}