[SW Expert Academy] - (D3)8931. 제로(Stack)

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AW5jBWLq7jwDFATQ&categoryId=AW5jBWLq7jwDFATQ&categoryType=CODE

SW Expert Academy

코드

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

public class Solution {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		
		int t = scan.nextInt();
		for(int tc=1; tc<=t; tc++) {
			int K = scan.nextInt();
			Stack<Integer> stack = new Stack<>();
			int sum = 0;
			
			for(int i=0; i<K; i++) {
				int num = scan.nextInt();
				
				// 0일경우 가장 마지막 수를 빼고, 0이 아닐경우 스택에 넣기
				if(num == 0) {
					stack.pop();
				} else {
					stack.push(num);
				}
			}
			
			// 스택의 모든 요소들 더하기.
			while(!stack.isEmpty()) {
				sum += stack.pop();
			}
			System.out.println("#" + tc + " " + sum);
		}
		
		scan.close();
	}

}

풀이

문제에서 핵심부분은 다음이다.

 

0을 외쳐서 가장 최근에 썼던 수를 지운다.

 

위 문장으로 스택을 유추할 수 있다.

문제에서 스택에 비어있는 상태에서 0이오는 상황은 안오는 것 같다.

따라서 0일때는 최근값을 지우고(pop), 0이 아닐때는 스택에 넣는다(push)

그 후 스택의 모든 요소들을 더하면 된다.

 

저번에는 시간초과가 발생했었는데, ArrayList로 풀고 요상하게 풀어서 시간초과가 발생했었다.