leetcode学习笔记

今天学习了leetcode上Hard级别的算法题，题干描述如下：

Merge k sorted linked lists and return it as one sorted list. Analyze and describe its complexity.

Example:

Input:
[
  1->4->5,
  1->3->4,
  2->6
]
Output: 1->1->2->3->4->4->5->6

如上所示，要把输入的链表进行重新排序并输出，这里要用到的数据结构有链表和队列，链表用于存储输入和输出以及中间结果，队列用于对输入的链表元素进行排序。这里采用的语言是java语言，对于列表，需要一个自定义类实现：

public class ListNode {
    int val;
    ListNode next;
    
    public ListNode(int val){
        this.val=val;
    }

}

java提供了Queue队列接口，该接口扩展了Collection接口，Queue使用时要尽量避免Collection的add()和remove()方法，而是要使用offer()来加入元素，使用poll()来获取并移出元素。它们的优点是通过返回值可以判断成功与否，add()和remove()方法在失败的时候会抛出异常。 如果要使用前端而不移出该元素，使用element()或者peek()方法。

值得注意的是LinkedList类实现了Queue接口，因此我们可以把LinkedList当成Queue来用。

如果要实现一定优先顺序的队列，例如最大最小堆，可以考虑使用PriorityQueue类，实例化该类需要实例化一个Comparator类来规定优先顺序，重写方法compare。

public static ListNode mergeKLists(List<ListNode> lists){
        Queue<ListNode> heap=new PriorityQueue(new Comparator<ListNode>(){
            @Override
            public int compare(ListNode l1, ListNode l2) {
                // TODO Auto-generated method stub
                return l1.val-l2.val;
            }
        });
        
        ListNode head=new ListNode(0),tail=head;
        for(ListNode list: lists){
            heap.offer(list);
        }
        
        while(heap!=null){
            tail.next=heap.poll();
            tail=tail.next;
            if(tail.next!=null){
                heap.offer(tail);
            }
        }
        
        return head.next;
    }