백준_11286 절댓값 힙(자바) / 자료구조

시간&메모리 제한

 

문제

 

입력&출력

 

문제 풀이

package com.Back;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;

public class Back_11286 {

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		StringBuilder sb = new StringBuilder();
		
		PriorityQueue<Integer> pq = new PriorityQueue<Integer>((a,b)->{
			if(Math.abs(a)==Math.abs(b)) {
				return a>b?1:-1;
			}
			return Math.abs(a)-Math.abs(b);
		}) ;
		
		for (int i = 0; i < N; i++) {
			int tmp = Integer.parseInt(br.readLine());
			if(tmp==0) {
				if(pq.isEmpty()) {
					sb.append("0\n");
				}else {
					sb.append(pq.poll()+"\n");
				}
			}else {
				pq.add(tmp);
			}
		}
		
		System.out.println(sb);
	}
}

 

※ 내 생각

자료구조를 이용해서 해결하는 문제입니다.
일단은 값을 비교하기 위해서 우선순위큐를 만들어서 해결을 했습니다.
우선순위 큐를 만들 때, 비교 조건을 넣어주면 값이 들어올 때, 인덱스 위치가 변하게 됩니다.
0인경우 비어있을 때, 0 출력을, 비어있지 않은 경우, 들어있는 값 출력을 해줍니다.
0이 아닌 경우 값을 우선순위 큐에 넣어줍니다.