題目要求:
Sort a linked list in?O(n?log?n) time using constant space complexity.
?數據結構定義:
1 /** 2 * Definition for singly-linked list. 3 * struct ListNode { 4 * int val; 5 * ListNode *next; 6 * ListNode(int x) : val(x), next(NULL) {} 7 * }; 8 */
這里用的數據結構是單鏈表。時間復雜度是O(n?log?n)的排序算法有:堆排序,快速排序,歸并排序。題目要求使用常量的空間復雜度,這三種排序方法的空間復雜度分別是O(n),O(1),O(n)[對于數組]。歸并排序的空間復雜度是O(n),因為對于數組,歸并排序需要額外的O(n)的空間來處理數組,而對于鏈表而言,操作都是對指針的,所以可以將空間復雜度降到O(1)。
這里使用歸并排序來實現代碼。
?
1 /** 2 * Definition for singly-linked list. 3 * struct ListNode { 4 * int val; 5 * ListNode *next; 6 * ListNode(int x) : val(x), next(NULL) {} 7 * }; 8 */ 9 class Solution { 10 public: 11 ListNode *sortList(ListNode *head){ 12 if (head == NULL) 13 return head; 14 else if (head->next == NULL) 15 return head; 16 17 ListNode *slow = head, *fast = head; 18 while (fast->next != NULL && fast->next->next != NULL) 19 { 20 slow = slow->next; 21 fast = fast->next->next; 22 } 23 24 fast = slow; 25 slow = slow->next; 26 fast->next = NULL; 27 28 fast = sortList(head); 29 slow = sortList(slow); 30 31 return mergeList(fast,slow); 32 33 } 34 ListNode* mergeList(ListNode* list1Head, ListNode* list2Head){ 35 if (list1Head == NULL) return list2Head; 36 if (list2Head == NULL) return list1Head; 37 38 ListNode* result, *temp; 39 // initialize the first node of the merged result list 40 if (list1Head->val < list2Head->val){ 41 result = list1Head; 42 list1Head = list1Head->next; 43 } 44 else 45 { 46 result = list2Head; 47 list2Head = list2Head->next; 48 } 49 // operate the result list with temp 50 temp = result; 51 // end merge if at least one of the two lists reaches the tail node 52 while (list1Head != NULL && list2Head != NULL){ 53 if (list1Head->val < list2Head->val){ 54 temp->next = list1Head; 55 list1Head = list1Head->next; 56 } 57 else{ 58 temp->next = list2Head; 59 list2Head = list2Head->next; 60 } 61 temp = temp->next; 62 } 63 64 if (list1Head != NULL) 65 temp->next = list1Head; 66 else 67 temp->next = list2Head; 68 69 return result; 70 } 71 };
這個簡單的OJ浪費了我大概半個多小時在調試BUG,實在有點太菜了。AC之前我的代碼遇到了下面幾個惡心的BUG。
1. 在找到鏈表的中間節點后,用head和slow來表示左右兩個子臉變的時候,沒有把中間節點的next指針置為空,導致程序無限遞歸,最后導致堆溢出。
2. 在合并排序數組的函數中,處理未添加到result鏈表中的節點(64~67)時,if語句條件寫錯了。....
.....
沉心學習,努力積累,希求爆發。
?