[Java/백준/스택] 10773 - 제로

문제

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

10773번: 제로

 

코드

내장 Stack 클래스를 사용했다.

그냥 숫자면 Stack에 push 해주고 0이면 pop을 해줬다.

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));
		Stack<Integer> stack = new Stack<Integer>();
		int N = Integer.parseInt(br.readLine());
		
		for(int i=0; i<N; i++) {
			int num = Integer.parseInt(br.readLine());
			
			if(num == 0)
				stack.pop();
			else
				stack.push(num);
		}

		int sum = 0;

		if(stack.isEmpty())
			System.out.println(sum);
		else {
			for(int element : stack) {
				sum+=element;
			}
			System.out.println(sum);
		}
	}	
}