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

 

이 문제는 간단한데 내부 반복문 부분을 어떻게 처리할까에 대한 고민 때문에 해결하는데 시간이 걸렸다. 이 부분은 내부 반복문을 따로 함수로 만들어서 호출하여 사용하는 방식으로 해결하였다.

일단 ( 나 [ 문자가 들어오면 스택에 저장하고, 이에 맞는 괄호가 들어오면 스택에서 제거한다. 이렇게 되었을 때 스택이 비어있으면 yes를 아니면 no를 출력해야 한다.

실수했던 부분 중 다른 부분은 (나 [가 들어오기전 닫히는 괄호가 생가는 부분이다. 이 점도 조건문을 통해 해결하였다.

https://ddamdoo.tistory.com/111

이 문제를 먼저 풀어보고 풀면 더 도움이 될 것 같다.

 

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

public class p4949 {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String s;
		
		for (;;) {
			s = br.readLine();
			if (s.equals(".")) {
				break;
			}
			System.out.println(func(s));
		}
	}
	
	public static String func(String s) {
		Stack<Character> st = new Stack<>();
		
		for (int i = 0; i < s.length(); i++) {
			if (s.charAt(i) == '(' || s.charAt(i) == '[') {
				st.push(s.charAt(i));
			} else if (s.charAt(i) == ')') {
				if (st.empty() || st.peek() != '(' ) {
					return "no";
				} else {
					st.pop();
				}
			} else if (s.charAt(i) == ']') {
				if (st.empty() ||st.peek() != '[') {
					return "no";
				} else {
					st.pop();
				}
			}
		}
		if (st.empty()) {
			return "yes";			
		} else {
			return "no";
		}
	}
}