[백준알고리즘] 4949번: 균형잡힌 세상 -Java

[백준알고리즘] 4949번: 균형잡힌 세상 -Java

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

4949번: 균형잡힌 세상

 

Stack문제이지만 문제의 조건들만 잘 맞춰준다면 어렵지 않은 문제이다.

 

 

일단 코드를 작성하기 전에 문제가 되는 경우들을 고려해봐야 한다.

1. 왼쪽 소괄호인데 오른쪽 대괄호가 오는 경우와

   왼쪽 대괄호인데 오른쪽 소괄호가 오는 경우

2. 왼쪽 소괄호나 왼쪽 대괄호 없이 오른쪽 소괄호나 오른쪽 대괄호가 나온 경우

 

즉 오른쪽 괄호를 넣어줄 때만 체크를 해주면 된다.

오른쪽 괄호를 넣을 때, 괄호가 없는 상태이면 안되고, 마지막에 있는 왼쪽 괄호가 자신과 짝인 괄호이어야 한다.

 

 

Stack으로 구성을 했고 오른쪽 괄호 존재가 잘못되어 no를 출력해야 하는 부분에서 stack.push를 해준 이유는 마지막에 "yes"와 "no"를 출력하는 부분에서 stack이 비어있는지로 판단을 할 건데, stack이 비어있는데 오른쪽 괄호가 와서 break를 해버리게 되면 그대로 "yes"가 출력되기 때문이다.

 

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

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int input_len, i;
		Stack<Character> stack = new Stack<Character>();
		char t;
		
		String input = br.readLine();
		while(!input.equals(".")) {
			stack.clear();
			
			input_len = input.length();
			for(i = 0; i < input_len; i++) {
				t = input.charAt(i);
				if(t == '(' || t == '[') {
					stack.push(t);
				}
				else if(t == ')' || t == ']') {
					if(stack.isEmpty() || (t == ')' && stack.peek() != '(') || (t == ']' && stack.peek() != '[')) {
						stack.push(t);
						break;
					}
					
					stack.pop();
				}
			}
			
			if(stack.isEmpty()) {
				System.out.println("yes");
			}
			else {
				System.out.println("no");
			}
			
			input = br.readLine();
		}
	}

}

 

 

잘못된 점이나 부족한 점 지적해주시면 감사하겠습니다