[백준] 10773번: 제로

 

https://blog.naver.com/zzang9ha?Redirect=Write&categoryNo=52

https://blog.naver.com/PostWriteForm.nhn?blogId=zzang9ha&Redirect=Write&categoryNo=52&redirect=Write&widgetTypeCall=true

코드

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

public class Main {

	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		Stack<Integer> stack = new Stack<Integer>();
		int K = scan.nextInt();
		int sum = 0;
		
		for(int i=0; i<K; i++) {
			int num = scan.nextInt();
			if(num == 0) 		// 입력받은 수가 0일경우
				stack.pop();	// 가장 마지막에 넣은 숫자 뺌
			else 				// 입력받은 수가 0이 아닐경우
				stack.push(num);// stack에 데이터 삽입
		}
		while(!stack.isEmpty()) // stack이 빌동안 반복
			sum += stack.pop();	// 합
		
		System.out.println(sum);
		scan.close();
	}
}

 

풀이

스택을 이용하면 쉽게 해결할 수 있다.

입력받은 수가 0일때, 가장 마지막에 넣은 데이터를 빼고

입력받은 수가 0이 아닐때 스택에 데이터를 넣으면 된다.

그 후 스택에 데이터가 존재할동안 합을 구한다.