[SW Expert Academy] 제로 (D3)

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AW5jBWLq7jwDFATQ&categoryId=AW5jBWLq7jwDFATQ&categoryType=CODE&problemTitle=&orderBy=PASS_RATE&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=5

SW Expert Academy

 

import java.util.ArrayDeque;
import java.util.Iterator;
import java.util.Scanner;

class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int t = sc.nextInt();

		for (int tc = 1; tc <= t; tc++) {
			int k = sc.nextInt();

			ArrayDeque<Integer> q = new ArrayDeque<>();

			for (int i = 0; i < k; i++) {
				int num = sc.nextInt();

				if (num == 0) {
					q.pollLast();
				} else {
					q.add(num);
				}
			}

			int sum = 0;

			Iterator<Integer> it = q.iterator();

			while (it.hasNext()) {
				sum += it.next();
			}

//			for (int num : q) {
//				sum += num;
//			}

			System.out.println("#" + tc + " " + sum);

		}

	}
}

 

iterator나 for each문으로 q안의 값들을 돌면서 계산해준다.