FilesExpand file tree

 main

/

P33_2.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

32 lines (28 loc) · 1.19 KB

 main

/

P33_2.java

Copy path

Top

File metadata and controls

• Code
• Blame

32 lines (28 loc) · 1.19 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

package ch11;

import java.util.HashMap;

import java.util.Map;

import java.util.PriorityQueue;

public class P33_2 {

public int[] topKFrequent(int[] nums, int k) {

// 각 엘리먼트의 빈도수를 저장할 해시맵 선언, 엘리먼트=>빈도수

Map<Integer, Integer> frequencyMap = new HashMap<>();

// 각 엘리먼트의 빈도수를 반복하며 계산하여 저장

for (int n : nums) {

// 처음 빈도수를 계산하는 엘리먼트는 기본값 0으로, 모두 +1 하여 저장

frequencyMap.put(n, frequencyMap.getOrDefault(n, 0) + 1);

}

// 빈도순으로 정렬되는 우선순위 큐 선언

PriorityQueue<int[]> pq = new PriorityQueue<>((a, b) -> b[1] - a[1]);

// 우선순위 큐에 각 엘리먼트와 빈도수 삽입

for (int elem : frequencyMap.keySet()) {

pq.add(new int[]{elem, frequencyMap.get(elem)});

}

// 결과로 리턴할 변수 선언

int[] result = new int[k];

// k번까지만 반복하여 우선순위 큐에서 결과 추출

for (int i = 0; i < k; i++) {

result[i] = pq.poll()[0];

}

return result;

}

}