[JAVA] 백준 11286 절댓값 힙

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

11286번: 절댓값 힙

 

- 절댓값 힙은 다음과 같은 연산을 지원하는 자료구조 

 

1. 배열에 정수 x(x != 0)을 넣는다.

2. 배열에서 절댓값이 가장 작은 값을 출력하고, 그 값을 배열에서 제거, 절댓값이 가장 작은 값이 여러개일 때는, 가장 작은 수를 출력하고, 그 값을 배열에서 제거한다. 

 

- 프로그램은 처음에 비어있는 배열에서 시작하게 된다. 

 

- x가 0이 아니라면 배열에 x라는 값을 넣는(추가하는) 연산

- x가 0이라면 배열에서 절댓값이 가장 작은 값을 출력하고 그 값을 배열에서 제거

- 만약 배열이 비어 있는 경우인데 절댓값이 가장 작은 값을 출력하라고 한 경우에는 0을 출력 

 

 

문제 풀이 

 

- Comparator 클래스를 만들어서 절댓값을 기준으로 우선 순위 큐 선언 

 

Comparator 

 

- 기본 정렬기준 외에 다른 기준으로 정렬하고자할 때 사용 

- Comparator<클래스 타입> 클래스명  = new Comparator <클래스 타입>()으로 정의 하고 compare 메소드를 오버라이드해서 사용 

- 클래스 타입 비교대상 2개를 지역 변수 o1,o2로 선언하고 return 값으로 음수,0,양수를 반환하면서 정렬할 수 있음

 

 

[JAVA]BOJ(백준) - 절댓값 힙 - 11286 (tistory.com)

[JAVA]BOJ(백준) - 절댓값 힙 - 11286

 

- 기존의 우선순위와는 다르게 큐를 만들면서 정렬을 할 때 두번의 정렬이 필요 

1) 절댓값 기준으로 한번 정렬

2) 절댓값이 같다면 오름차순 정렬 

 

 

정답 

 

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

public class Main {
    public static void main(String[] args) throws IOException
    {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        
        AbsComparator abscomparator = new AbsComparator();
        //우선순위가 낮은 순 = 최소힙
        PriorityQueue<Integer> pq = new PriorityQueue<>(abscomparator);
        
        
        int N = Integer.parseInt(br.readLine());
        
        StringTokenizer st;
        for(int i=0; i<N; i++)
        {
            st = new StringTokenizer(br.readLine());
            int value = Integer.parseInt(st.nextToken());
            
            //값 제거 
            if(value == 0)
            {
                if(pq.isEmpty())
                {
                    sb.append("0").append("\n");
                }
                else
                {
                    int ans = pq.poll();
                    sb.append(ans).append("\n");
                }
            }
            //값 추가 
            else
            {
                pq.add(value);
            }
        }
        
        System.out.println(sb);
    }
}

//우선순위 비교 클래스 생성 
class AbsComparator implements Comparator<Integer>
{
	@Override
	public int compare(Integer o1, Integer o2)
	{
		Integer a = Math.abs(o1);
		Integer b = Math.abs(o2);
		
		if(a>b)
		{
			return 1;
		}
		else if(a<b)
		{
			return -1;
		}
		else //절댓값이 같다면 
		{
			if(o1>o2)
			{
				return 1;
			}
			else if(o1<o2)
			{
				return -1;
			}
			else
			{
				return 0;
			}
		}
	}
}