两个有序链表的合并
对于两个有序链表合并成一个新的有序链表的代码很简单,但是要注意一些测试用例:
比如说一个为空链表;两个链表不一样长,肯定最后有一个链表要单独的挂到结果链表中去的。
ListNode* mergeTwoLists(ListNode* l1, ListNode* l2) { if(l1 == NULL) return l2; if(l2 == NULL) return l1; ListNode dummy(-1); ListNode *p = &dummy; while(l1 != NULL && l2 != NULL) { if(l1->val < l2->val) { p->next = l1; p = p->next; l1 = l1->next; p->next = NULL; } else { p->next = l2; p = p->next; l2 = l2->next; p->next = NULL; } }//while p->next = l1 == NULL ? l2 : l1; return dummy.next; }
注意程序中用到了一个头结点,这个头结点就是为了链表插入的方便,真正返回的时候还是放回的dummy.next。
K个有序链表的合并
这个问题当然可以利用上面的合并两个有序链表的方法,把前两个合并成一个,然后再同第三个合并,这样一次下去。但是这让我们想起来归并算法,把非常小的子问题首先合并,然后再合并中等的问题,最后再把中等的问题合并成大的问题。这样的时间复杂度要比一次合并小得多。所以利用归并的思想,设计代码如下:
ListNode* mergeKLists(vector<ListNode*>& lists) { int n = lists.size(); if(n == 0) return NULL; while(n > 1) { int k = (1 + n) / 2; for(int i = 0; i < n / 2; i++) { lists[i] = mergeTwoLists(lists[i], lists[i + k]); } n = k; } return lists[0]; }
这里利用归并的时候,把整个的问题划分还是有一定的技巧的。
int k = (n + 1) / 2; 1.如果n是偶数,那么k是后半部分的开始,并且前面直接就是前半部分。 2.如果n是奇数,那么k是后半部分的开始,但是前面隔着一个n/2这个元素, 再往前才是对应的前半部分。