[백준 알고리즘 자바] 10773 : 제로

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

10773번: 제로

---

문제

• 나코더 기장 재민이는 동아리 회식을 준비하기 위해서 장부를 관리하는 중이다.
• 재현이는 재민이를 도와서 돈을 관리하는 중인데, 애석하게도 항상 정신없는 재현이는 돈을 실수로 잘못 부르는 사고를 치기 일쑤였다.
• 재현이는 잘못된 수를 부를 때마다 0을 외쳐서, 가장 최근에 재민이가 쓴 수를 지우게 시킨다.
• 재민이는 이렇게 모든 수를 받아 적은 후 그 수의 합을 알고 싶어 한다. 재민이를 도와주자!

입력

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

출력

• 재민이가 최종적으로 적어 낸 수의 합을 출력한다. 최종적으로 적어낸 수의 합은 231-1보다 작거나 같은 정수이다.

입출력 예제

---

풀이방식

이번 문제는 Stack을 사용하여 푸는 문제이다.

---

설계과정

1. 스택에 0이 아닌 정수를 넣는다.

2. 0이 입력되면 스택에서 값을 뺀다.

3. 스택에 남아있는 값들을 더한다.

---

풀이과정

1. 값을 입력받는 코드를 작성한다.

 

2. 메소드를 생성하고 스택을 선언한다.

 

3. 0이 아닌 정수가 들어오면 스택에 값을 넣고, 0이 들어오면 스택에서 값을 뺀다.

• push() : 스택에 값을 추가한다.
• pop() : 스택에서 값을 제거한다.

4. 스택에 남아있는 값들을 전부 더한다.

• isEmpty() : 값이 비었는지 확인하여 true / false로 리턴한다.

---

답안소스

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

public class Main {

	public static int allSum(int K, int[] arr) {
		int result = 0;
		Stack<Integer> stack = new Stack<>();
		
		for(int i=0; i<K; i++) {
			if(arr[i] == 0) {
				stack.pop();
			}else {
				stack.push(arr[i]);
			}
		}
		
		while(!stack.isEmpty()) {
			result += stack.pop();
		}
		
		return result;
	}
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int K = sc.nextInt();
		int[] arr = new int[K];
		
		for(int i=0; i<K; i++) {
			arr[i] = sc.nextInt();
		}
		
		System.out.println(Main.allSum(K, arr));
	}
}