알고리즘 공부/hashmap , 자료구조
백준 1927번 with Kotlin
_우지
2021. 8. 6. 16:19
최소 힙
시간 제한메모리 제한제출정답맞은 사람정답 비율
1 초 (추가 시간 없음) (하단 참고) | 128 MB | 29638 | 13365 | 10415 | 48.323% |
문제
널리 잘 알려진 자료구조 중 최소 힙이 있다. 최소 힙을 이용하여 다음과 같은 연산을 지원하는 프로그램을 작성하시오.
- 배열에 자연수 x를 넣는다.
- 배열에서 가장 작은 값을 출력하고, 그 값을 배열에서 제거한다.
프로그램은 처음에 비어있는 배열에서 시작하게 된다.
입력
첫째 줄에 연산의 개수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가 0이라면 배열에서 가장 작은 값을 출력하고 그 값을 배열에서 제거하는 경우이다. 입력되는 자연수는 231보다 작으며, 음의 정수는 입력으로 주어지지 않는다.
출력
입력에서 0이 주어진 회수만큼 답을 출력한다. 만약 배열이 비어 있는 경우인데 가장 작은 값을 출력하라고 한 경우에는 0을 출력하면 된다.
예제 입력 1 복사
9 0 12345678 1 2 0 0 0 0 32
예제 출력 1 복사
0 1 2 12345678 0
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | import java.io.BufferedReader import java.io.BufferedWriter import java.io.InputStreamReader import java.io.OutputStreamWriter import java.lang.Math.pow import java.lang.NumberFormatException import java.util.* import kotlin.math.abs import kotlin.math.pow import kotlin.math.sqrt val br = BufferedReader(InputStreamReader(System.`in`)) fun main()=with(br){ val bw = BufferedWriter(OutputStreamWriter(System.out)) val n = readLine()!!.toInt() val minHeap = PriorityQueue<Int>() for(i in 0 until n){ val a = readLine()!!.toInt() if(a==0){ if(!minHeap.isEmpty()){ bw.write("${minHeap.poll()}\n") } else bw.write("0\n") } else{ minHeap.add(a) } } bw.flush() bw.close() } | cs |