본문 바로가기

코딩테스트/백준

[백준-자바] 10773번 제로 / 2022.03.15

 

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

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

 

 

import java.util.Scanner;
import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        Stack<Integer> stack = new Stack<>();

        int K = sc.nextInt();

        for (int i = 0; i < K; i++) { // K번 만큼 반복
            int num = sc.nextInt(); // 수 입력
            if (num == 0) { // 입력받은 수가 0 이라면
                stack.pop(); // 맨 위의 수를 pop
            } else { // 0이 아니라면
                stack.push(num); // 입력받은 수를 push
            }
        }

        int hap = 0; // 스택에 남아있는 요소의 합 구하기
        while (!stack.isEmpty()) { // 스택이 빌 때까지
            hap += stack.pop(); // pop한 요소를 hap변수에 더해줌
        }

        System.out.println(hap); // 결과 출력
    }
}