[백준] 10799. 쇠 막대기 (Java) [Stack]

문제

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

10799번: 쇠막대기

 

 

사용한 알고리즘

Stack

 

 

풀이

괄호에서 레이저를 파악하는 것이 핵심인 문제였다.

Stack을 통해 '('을 삽입을 하면서 '()'이 있는 경우는 레이저로 판단하여, 조건 문으로 분기를하여 풀이를 하였다.

 

자세한 것은 풀이를 확인해보자.

 

 

코드

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

public class Test_Stack {
    static StringBuffer sb;
    static int ans = 0;
    public static void main(String[] args) {
        //입력
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        sb = new StringBuffer();
        try {
            sb.append(bf.readLine());
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        go();
    }

    static void go(){
        Stack<Character> stack = new Stack<>();

        for (int i = 0; i < sb.length(); i++) {
            if (sb.charAt(i) == '(') {
                stack.add(sb.charAt(i));
            } else if (sb.charAt(i) == ')') {
                //레이저가 있는경우 "()"
                if (i >= 1 && sb.charAt(i - 1) == '(') {
                    stack.pop();
                    ans += stack.size(); //현재 stack의 사이즈만 큼 더해준다/
                } else { //그 전 괄호가 닫힌 괄호면 레이저가 아님
                    ans++; //단순히 1을 더해줌
                    stack.pop();
                }
            }
        }
        System.out.println(ans);
    }
}

 

Stack 사용법

https://coding-factory.tistory.com/601

[Java] 자바 Stack 클래스 사용법 & 예제 총정리