标签:object compare ++ 合并 对象 contain val rem nta
一、题目
1、审题
2、分析
将 n 个有序链表合并成为一个有序链表。
二、解答
1、思路:
Java 有一个 PriorityQueue,该队列能够将放入的对象进行排序。故将所有非空队头元素放入该队列,取出一个元素后,再放入所取出元素指向的下一个非空结点;当队列中元素为空时,取出的顺序即为排序后的新的链表的元素顺序。
有关 PriorityQueue:
add(E e) 添加元素
clear() 清空
contains(Object o) 检查是否包含当前参数元素
offer(E e) 添加元素
peek() 读取元素,(不删除)
poll() 取出元素,(删除)
remove(Object o) 删除指定元素
size() 返回长度
/** * Definition for singly-linked list. * public class ListNode { * int val; * ListNode next; * ListNode(int x) { val = x; } * } */ class Solution { public ListNode mergeKLists(ListNode[] lists) { if(lists == null || lists.length == 0) return null; /* * PriorityQueue: 自动根据优先级排序的队列 */ PriorityQueue<ListNode> heap = new PriorityQueue<ListNode>(10, new Comparator<ListNode>() { @Override public int compare(ListNode o1, ListNode o2) { return o1.val - o2.val; } }); for (int i = 0; i < lists.length; i++) if(lists[i] != null) heap.offer(lists[i]); // add ListNode head = null; ListNode pre = head; while(heap.size() > 0) { ListNode cur = heap.poll(); // 取出,删除 if(head == null) head = cur; else pre.next = cur; pre = cur; if(cur.next != null) heap.offer(cur.next); //add } return head; } }
标签:object compare ++ 合并 对象 contain val rem nta
原文地址:https://www.cnblogs.com/skillking/p/9417428.html