[백준 11286] - 절댓값 힙

JIWOO YUN·2023년 4월 1일
0
post-custom-banner

문제링크

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

구현방법

연산의 개수가 10만개이기 때문에 O(N^2)을 넘기면 안되고 그렇기 때문에 우선순위큐를 통해서 값이 들어왔을경우 바로 정리가 되게 진행했습니다.

우선순위큐의 시간복잡도는 O(logN)이기 때문에 시간초과가 걸리지 않기 때문입니다.

우선순위큐에 넣을 때 comparater를 통해 재정의를 진행해서 현재 문제가 제시한 조건을 맞춥니다.

만약 절댓값이 같으면 그 숫자끼리 비교해서 오름차순정렬을 진행하고 아닌경우 절댓값을 기준으로 오름차순으로 정렬합니다.

구현알고리즘

우선순위 큐(priority Queue)

CODE

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader((System.in)));
        //절댓값 기준으로 우선순위 정렬 
        PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                //둘의 절댓값이 같으면 숫자를 비교해서 오름차순 정렬
                if(Math.abs(o1) == Math.abs(o2))
                {
                    return o1 - o2;
                    //그외는 절댓값 기준 정렬
                }else
                {
                    return Math.abs(o1) - Math.abs(o2);
                }
            }
        });

        StringBuilder sb = new StringBuilder();
        int T = Integer.parseInt(br.readLine());
        for(int test= 0; test< T;test++){
            int number = Integer.parseInt(br.readLine());
            //값이 0이 입력된경우 가장 작은 수를 출력하고 다음 값으로 이동 or 큐에 없는 경우 0 실어주고 다음값으로
            if(number == 0){
                if(pq.isEmpty()){
                    sb.append(0).append("\n");
                }
                else {
                    sb.append(pq.poll()).append("\n");
                }
            }else{
                pq.offer(number);
            }
        }

        System.out.println(sb);
    }
}
profile
열심히하자
post-custom-banner

0개의 댓글