BOJ_S4_4949

📁 [S4_4949] 균형잡힌 세상

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();

		boolean check = true;
		while (check) {
			String str = br.readLine();

			// 종료 조건
			if (str.equals(".")) {
				check = false;
			} else {
				String res = check(str);
				sb.append(res).append("\n");
			}
		}
		System.out.println(sb);
	}

	// Stack을 이용해 문자가 균형잡힌 문자열인지 체크
	static String check(String str) {
		String res = "yes";
		Stack<Character> stack = new Stack<>();
		for (int i = 0; i < str.length(); i++) {
			// 시작부분
			if (str.charAt(i) == '[' || str.charAt(i) == '(') {
				stack.add(str.charAt(i));
			}
			// 대괄호 닫힘
			if (str.charAt(i) == ']') {
				if (stack.size() == 0) {
					res = "no";
					break;
				}
				char pop = stack.pop();
				if (pop != '[') {
					res = "no";
					break;
				}
			}

			// 소괄호 닫힘
			if (str.charAt(i) == ')') {
				if (stack.size() == 0) {
					res = "no";
					break;
				}
				char pop = stack.pop();
				if (pop != '(') {
					res = "no";
					break;
				}
			}
		}

		// 마무리가 안된경우 no
		if(stack.size()>0) res="no";
		return res;
	}
}

🤔 나의 생각

Stack을 사용하는 문제이다.

• ’[’ 과 ‘(‘ 이면 stack에 add
• stack에 아무것도 없는데 닫힐 경우 no
• ’]’ 일때 stack에서 pop 한 것이 ‘[’ 이면 통과, 아니면 no
• ’)’ 일때 stack에서 pop 한 것이 ‘(‘ 이면 통과, 아니면 no
• 마지막에 stack에 값이 남아있으면 no ( 완성이 안된 경우 )

빨리 문제를 풀고 스택에 아무것도 없는 경우와 마지막 완성이 안된 경우를 놓쳐서 시간이 좀 걸렸다.