FilesExpand file tree

 main

/

P27_1.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

41 lines (34 loc) · 1.19 KB

 main

/

P27_1.java

Copy path

Top

File metadata and controls

• Code
• Blame

41 lines (34 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

33

34

35

36

37

38

39

40

41

package ch10;

import datatype.ListNode;

import java.util.PriorityQueue;

public class P27_1 {

public ListNode mergeKLists(ListNode[] lists) {

PriorityQueue<ListNode> pq = new PriorityQueue<>((o1, o2) -> {

// 동일한 값이면 무시

if (o1.val == o2.val)

return 0;

// 새로운 값(o1)이 기본 값(o2)보다 크면 뒤에 위치

else if (o1.val > o2.val)

return 1;

// 작으면 앞에 위치

else

return -1;

});

ListNode root = new ListNode(0);

ListNode tail = root;

// 각 연결 리스트의 첫 번째 노드를 큐에 저장

for (ListNode node : lists)

if (node != null)

pq.add(node);

// 큐가 모두 빌때까지 반복

while (!pq.isEmpty()) {

// 우선순위에 따라 추출, 다음 노드로 이동

tail.next = pq.poll();

tail = tail.next;

// 추출한 연결 리스트의 다음 노드는 다시 큐에 저장

if (tail.next != null)

pq.add(tail.next);

}

return root.next;

}

}