알고리즘 문제풀이/백준
[BOJ][JAVA] 10773 - 제로
쑤테크
2021. 9. 22. 20:21
문제 출처 : https://www.acmicpc.net/problem/10773
10773번: 제로
첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경
www.acmicpc.net
문제
설명
풀이 방법으로 스택(FILO)을 쉽게 떠올릴 수 있는 문제입니다. 입력이 0일 때마다 스택을 pop하고, 그렇지 않으면 그 수를 스택에 넣습니다. 입력이 끝나면, 스택이 빌 때까지 pop하면서 pop한 수를 결과에 누적시켜 더하면 되는, 비교적 간단한 문제였습니다.
코드
import java.util.Scanner;
import java.util.Stack;
public class BOJ10773_Zero {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
Stack<Integer> stack = new Stack<>();
int sum=0;
int num=sc.nextInt();
for(int i=0; i<num; i++) {
int input = sc.nextInt();
if(input==0)
stack.pop();
else {
stack.push(input);
}
}
while(!stack.isEmpty()) {
sum += stack.pop();
}
System.out.println(sum);
}
}