[알고리즘] 백준 - 10799번 : 쇠막대기

알고리즘 문제 출처 : https://www.acmicpc.net/problem/10799

10799번: 쇠막대기

 

내가 작성한 코드

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

public class Main {
	
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		String work = br.readLine();
		work = work.replace("()", "*");
		Stack<Character> st = new Stack<>(); 
		int answer = 0;
		for(int i=0; i<work.length(); i++) {
			if(work.charAt(i)=='(') { //여는 괄호일 때 스택에 넣는다.
				st.push('(');
			}
			if(work.charAt(i)==')') { //닫는 괄호일 때 스택에 있는 여는 괄호 삭제
				st.pop();
				answer++; //레이저로 자르면 (레이저 개수+1)조각 나옴
			}
			if(work.charAt(i)=='*') { //레이저를 만났을 때 스택에 있는 여는 괄호 개수 세기
				answer += st.size();
			}
		}
		bw.write(Integer.toString(answer));
		bw.flush();
		bw.close();
	}
}