백준 10773 제로 (Java)

1. 문제 링크

10773번: 제로

 

2. 문제 및 입출력예제

 

3. 문제 풀이

입력받은 수를 stack에 push고, 0을 입력 받을 때 pop 한다.

모든 입력이 끝나면 stack에서 pop하여 모든 수를 더한 값을 출력한다.

 

4. 코드

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


public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		Stack<Integer> stack = new Stack<>();
		int K = Integer.parseInt(br.readLine());
		for(int i=0;i<K;i++) {
			int num = Integer.parseInt(br.readLine());
			if(num!=0) {
				stack.push(num);
			}
			else {
				stack.pop();
			}
		}
		int result = 0;
		int size = stack.size();
		for(int i=0; i<size;i++) {
			result+= stack.pop();
		}
		System.out.println(result);
	}
}