LeetCode第[21][23]题(Java)：Merge Sorted Lists

题目：合并两个已排序链表

难度：Easy

题目内容：

Merge two sorted linked lists and return it as a new list. The new list should be made by splicing together the nodes of the first two lists.

翻译：

合并两个已排序的链表，并将其作为一个新链表返回。新的链表应该通过将前两个列表的节点拼接在一起。

Example:

Input: 1->2->4, 1->3->4
Output: 1->1->2->3->4->4

我的思路：数据结构——已经指定链表，且不需要特殊操作，就用链表；

　　　　　算法——对一个链表进行遍历，如果链表2节点的值小于当前节点，就插入到它前面。

　　　　　　　　因为是单链表，所以需要前驱指针一起移动才能进行操作，并且还需要一个头节点作为返回用。

我的代码：

     public ListNode mergeTwoLists(ListNode l1, ListNode l2) {
         if (l1 == null || l2 == null) {
             return l1 == null ? l2 : l1;
         }

         ListNode head = new ListNode(0);
         ListNode pre = head;
         head.next = l1;

         while (l1 != null && l2 != null) {
             if (l2.val < l1.val) {
                 pre.next = l2;
                 l2 = l2.next;
                 pre.next.next = l1;

                 pre = pre.next; // pre归位
                 continue;   // 插入结束，l1不需要移动
             }
             pre = l1;
             l1 = l1.next;
         }
         if (l2 != null) {
             pre.next = l2; // l2 还有剩的话说明l1已经为null，直接接在pre后面即可
         }
         return head.next;
     }

我的复杂度：  时间： O（M + N）  空间： O（1）

编码过程出现问题：

1、在插入l2之后此时l1 是不需要移动的——>continue   出错用例：[5]、[1,2,4]

2、插入后，continue之前，还得把pre往前移动   出错用例：[5]、[1,2,4]

参考答案代码：

      public ListNode mergeTwoLists(ListNode l1, ListNode l2){
          if(l1 == null) return l2;
          if(l2 == null) return l1;
          if(l1.val < l2.val){
              l1.next = mergeTwoLists(l1.next, l2);
              return l1;
          } else{
              l2.next = mergeTwoLists(l1, l2.next);
              return l2;
          }
      }

答案复杂度：　　时间：O（N+M）    空间：O（N+M）

答案思路：

利用递归的思想，每次比较之后，将小的那个节点的后继改为下一次递归返回的节点，然后返回自己。所以最后会返回l1与l2中小的那一个节点，而它的后继是一条完整的由小到大的节点。

比较：我的那个方法空间复杂度比较好，而且由于没有使用递归运行相对较快。而答案方法则简洁明了。

---

---

补充：

23题：Merge k Sorted Lists

---

---

题目：就是21题的两个链表改成了k个

难度：Hard

我的思路：选取节点的思路肯定还是一样，选取k个链表表头中最小的那一个作为下一个节点，但是由于是k个所以在最内层也就是lists[]某一个或某几个为null的时候，就不能像两个链表一样直接返回另外一个那么简单了，就需要把原来lists中的null都给删除，但是由于是数组，所以操作起来比较麻烦得借用list。

这种类似于多排序数组归并，其实最佳选择是堆排序，不过我嫌麻烦就直接顺序搜索最小值了。

我的代码：

 class Solution {
     public ListNode mergeKLists(ListNode[] lists) {
         if (lists.length == 0) {
             return null;
         }

         //***   去除null节点   **//
         List<ListNode> l = new ArrayList<ListNode>(Arrays.asList(lists));
         boolean hasNull = l.remove(null);
         lists = l.toArray(new ListNode[l.size()]);
         //  end  //

         if (hasNull) {
             return mergeKLists(lists);
         }

         ListNode min = getMinNode(lists);
         min.next = mergeKLists(lists);

         return min;
     }

     public static ListNode getMinNode(ListNode[] lists) {
         int minIndex = 0;
         for (int i = 1; i < lists.length; i++) {
             if (lists[i].val < lists[minIndex].val) {
                 minIndex = i;
             }
         }

         ListNode minNode = lists[minIndex];
         lists[minIndex] = lists[minIndex].next; // min取出后，lists要更新

         return minNode;
     }
 }

结果：最后一两个用例栈溢出

我的复杂度： 时间：O （N*M）    N是lists的长度，M是lists中最长的那一个

　　　　　    空间：超级大。。。（因为每一次的去除null，都会新建一个lists的对应的ArrayList，并且toArray又新建了一个对应的Array），所以栈溢出了

优化：

那就用简单粗暴的方法去除null：

// 替换去除null //
        for (int i = 0; i < lists.length; i++) {
            if (lists[i] == null) {
                ListNode[] temp = new ListNode[lists.length-1];
                for (int j = 0; j < temp.length; j++) {
                    if (j < i) {
                        temp[j] = lists[j];
                    } else {
                        temp[j] = lists[j+1];
                    }
                }
                hasNull = true;
                lists = temp;
                i--;
            }
        }
// end//

这样一来就只多建了一个temp的Array。

【注意】：最后在数组删除元素之后不要忘了  i -- ，因为当前的i已经被删除了，最新的lists第i个已经是以前的第i+1个，如果继续循环那么将直接 i ++ 就会跳过之前的那个 i + 1。

优化后结果：不溢出了。。。但是运行超时了。

好吧看来还是得用堆排序——优先队列。

答案代码：

 class Solution {
     public ListNode mergeKLists(ListNode[] lists) {
         ListNode res=null;

         System.out.println("lists.length "+ lists.length);
         if(lists == null || lists.length== 0)
             return res;

         PriorityQueue pq = new PriorityQueue();
         for(int i=0; i<lists.length; i++){
             if(lists[i]!=null){
                 pq.offer(new PQNode(lists[i].val, 0, i));
                 lists[i] = lists[i].next;
             }
         }

         for(int i=0; !pq.isEmpty(); i++){
             PQNode n = (PQNode)pq.poll();
             int value = n.value;
             int index = n.index;
             int listNumber = n.listNumber;
             if(res==null){
                 res = new ListNode(value);
             }
             else{
                 ListNode temp = res;
                 while(temp.next!=null){
                     temp=temp.next;
                 }
                 temp.next = new ListNode(value);
             }
             if(lists[listNumber]!=null){
                 pq.offer(new PQNode(lists[listNumber].val, 0, listNumber));
                 lists[listNumber] = lists[listNumber].next;

             }
         }
         return res;
     }
 }

 class PQNode implements Comparable<PQNode>{
     int value;
     int index;
     int listNumber;

     public PQNode(int value, int index, int listNumber){
         this.value = value;
         this.index = index;
         this.listNumber = listNumber;
     }

     public int compareTo(PQNode obj){
         if(this.value < obj.value) return -1;
         if(this.value > obj.value) return 1;
         return 0;
     }
 }

答案复杂度： 时间：O （N*logM + M）   空间：O（M）

答案思路：思路没变，不过就是在取k个数组中最小的时候使用了一个优先级队列——最开始将k个链表头取出建立一个优先级队列（java.util.PriorityQueue），然后每次poll

出队头后再放进此队头原来所在链表的链表头，当发现此表头为null的时候，则优先级队列的大小减一。直到队列为空。