[백준알고리즘] 10773번: 제로 -Java

[백준알고리즘] 10773번: 제로 -Java

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

10773번: 제로

 

stack문제이다. 마침 Java 연습하고 있었는데 개이득 봤다. 좋았다!

 

 

Java에서는 java.util에서 Stack을 제공하고 있어서 편하게 코드를 짤 수 있었다.

 

문제는 사실 처음 봤을 때에는 배열을 이용할까 하다가 분류가 Stack이니 만큼 stack을 사용하기로 했다.

 

0이 나올 때 지울 수가 항상 존재함을 보장한다고 했으니 굳이 조건을 저렇게까지 써주지 않아도 되지만 써줬다!

 

오랜만에 자바 연습한 걸로 만족해야 될 것 같다.

 

 

import java.io.*;
import java.util.Stack;

public class Main {

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int i, t, sum = 0;
		
		/* input */
		int N = Integer.parseInt(br.readLine());
		Stack<Integer> stack = new Stack<Integer>();
		
		/* stack */
		for(i = 0; i < N; i++) {
			t = Integer.parseInt(br.readLine());
			
			if(t == 0 && !stack.isEmpty()) {
				stack.pop();
			}
			else if(t != 0){
				stack.push(t);
			}
		}
		
		/* sum */
		while(!stack.isEmpty()) {
			sum += stack.pop();
		}
		
		System.out.println(sum);
	}

}

 

 

잘못된 점이나 부족한 점 지적해주시면 감사하겠습니다