[java] 백준 10773번 제로

 

해당 문제 같은 경우 Stack 클래스를 사용하면 쉽게 풀 수 있다.

 

Stack은 LIFO구조로 나중에 들어온 값이 먼저 나간다. 힌트에서 볼 수 있는 결과와 같다.

push메소드와 pop메소드로 값을 넣고 뺄 수 있다.

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();
        int tot =0;
        for(int i=0; i<K; i++){
            int n = sc.nextInt();

            if (n==0)
                stack.pop();
            else
                stack.push(n);

        }
        for (int i : stack) {
            tot += i;
        }

        System.out.println(tot);
    }
}

 

K개수만큼 입력받아 입력받은 n이 0일 때만 현재 스택 구조에서 이전에 넣었던 값을 빼주도록 하면 된다.

 

그리고 그 후에 스택 구조에 남아있는 값들의 합을 구해주면 쉽게 풀 수 있는 문제다.