[Java] 백준 4949 : 균형잡힌 세상

🔗 문제 내용

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

4949번: 균형잡힌 세상

🌱 문제 풀이 방법

자료구조 시간에 스택 배운걸 떠올리면서 풀었다.

https://abcodef.tistory.com/11?category=903621 

[Java] 백준 9012 : 괄호

9012 괄호 문제와 유사해 그 코드를 활용했더니 깔끔하지 않아서 수정할 예정이다.

🤔 공부가 필요한 부분

더 깔끔하게 코드 작성하기

문제 제대로 안봐서 yes, no가 아닌 YES, NO로 출력해서 한번 틀렸음ㅠ

💻 코드

// 수정 예정
import java.util.Scanner;
import java.util.Stack;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		Stack<Character> stack = new Stack<>();

		while (true) {
			String str = sc.nextLine();
			if (str.equals(".")) {
				break;
			}
			String answer = "yes";
			for (int r = 0; r < str.length(); r++) {
				if (str.charAt(r) == '(' || str.charAt(r) == '[') {
					stack.push(str.charAt(r));
				} else if (str.charAt(r) == ')') {
					if (!stack.isEmpty()) {
						if (stack.peek() == '(') {
							stack.pop();
						} else
							answer = "no";
					} else
						answer = "no";
				} else if (str.charAt(r) == ']') {
					if (!stack.isEmpty()) {
						if (stack.peek() == '[') {
							stack.pop();
						} else
							answer = "no";
					} else
						answer = "no";
				}
			}

			if (!stack.isEmpty())
				answer = "no";
			stack.clear();
			System.out.println(answer);
		}
		sc.close();
	}
}