백준 11279번: 최대 힙 with Java

최대 힙

 

시간 제한메모리 제한제출정답맞힌 사람정답 비율

1 초 (추가 시간 없음) (하단 참고)

256 MB

82835

39697

31435

49.500%

문제

널리 잘 알려진 자료구조 중 최대 힙이 있다. 최대 힙을 이용하여 다음과 같은 연산을 지원하는 프로그램을 작성하시오.

1. 배열에 자연수 x를 넣는다.
2. 배열에서 가장 큰 값을 출력하고, 그 값을 배열에서 제거한다.

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

입력

첫째 줄에 연산의 개수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가 0이라면 배열에서 가장 큰 값을 출력하고 그 값을 배열에서 제거하는 경우이다. 입력되는 자연수는 231보다 작다.

출력

입력에서 0이 주어진 횟수만큼 답을 출력한다. 만약 배열이 비어 있는 경우인데 가장 큰 값을 출력하라고 한 경우에는 0을 출력하면 된다.

예제 입력 1 복사

13
0
1
2
0
0
3
2
1
0
0
0
0
0

예제 출력 1 복사

0
2
1
3
2
1
0
0

 

 

 

---

 

 

개강이 한 달도 안 남았다는 거 실화.....? 돌려조요 전 한 게 없어요 진짜루.

제발제발제발 7월 15일로만 돌려주면 정말. 열심히. 살겠씁니다. 정처기도 공부하고 영어 공부도 할게요 제발 돌려주세요...

8월 10일 실화인가

글고

실버2로 올라가서 그런가

문제도

개

어

려

워

.

.

.

 

자료구조 이제 뭔가 까마득해졌어. 힙을 아는데 모르는 상태가 되어버렸다. What is 힙 mean? ..이대로면 진자로 갓수가 되고 만다.

 

?

뭐야 내 글 다 날아갔어 진심 티스토리는 반성해라 작성하던 글이 자동으로 날아가는 건 도대체 뭔 경우냐고.. 왜 자꾸 지 혼자 로그아웃 되는데

힙이 미묘하게 기억 안 나서 힙 정의랑 최대힙 정의 갖다 써놨는데 장난해?? 아아악앙아아악 다시 쓰기 귀찮네요. 대충 쓰자면 최대힙 완전이진트리 중에 부모노드가 자식 노드보다 큰 게 최대힙이었습니다^^

 

첨엔 ArrayList로 작성하려고 했는데(걍 들어오는대로 입력 받다가 0 들어오면 최대값 ㅇ보내주기..) 찾다가 PriorityQueue 요걸 찾아서 사용했고 기본이 최소힙이어서 최대힙을 쓰려면 리버스해줘야한다는 걸 알게 되었습니다. 글고 if else 문만 잘 써주면 풀리는 생ㄱ각보다 쉬운 문제였음!(물론 이건 PriorityQueue를 사용했기 때문이겠져)

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Collections;
import java.util.PriorityQueue;
public class BJ_11279 {
	public static void main(String[] args) throws NumberFormatException, IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        
        // 최대 힙 쓰려면 reverseOrder로 써야 함
        PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Collections.reverseOrder());
        
        for (int i = 0; i < N; i++) {
            int x = Integer.parseInt(br.readLine());
            
            // 0 입력 들어오면 배열에서 가장 큰 값 출력하고 그 값 배열에서 제거
            if (x == 0) {
                // 힙 비어있으면 0 출력
                if (maxHeap.isEmpty()) {
                    System.out.println(0);
                } 
                else {
                    // 있으면 최대값 출력 후 제거
                    System.out.println(maxHeap.poll());
                }
            } 
            else {
                // x 힙에 추가
                maxHeap.add(x);
            }
        }
	}
}

 

 

성공!!!

휴... 다행히 시간 안에 해냈따.... 담 거도 지금 해야겟으요....