[Java] 백준 10773번 문제 (제로)

문제설명

 

소스코드

import java.util.Stack;
import java.util.Scanner;
public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		Stack<Integer> stack = new Stack<Integer>(); //스택 생성
		int K = sc.nextInt();
		for(int i = 0; i < K; i++) {
			int input = sc.nextInt();
			if(input == 0) stack.pop(); //0이면 팝
			else stack.push(input); //0이 아니면 푸시
		}
		int sum = 0;
		for(int i = 0; i < stack.size(); ++i) {
			sum += stack.elementAt(i); //스택의 모든 원소를 더함
		}
		System.out.println(sum);		
	}
}

 

설명

• 컬렉션 프레임워크의 Stack을 사용하였다.