[leetcode]Sort List
merge sort,记了长度。如果断开链表时把第一个链表的最后一个节点的next设为NULL,就不用在函数参数里传长度了。一直错误是因为在生成的链表最后的next没有设成NULL。用了dummy node很有用。
可以借鉴http://blog.csdn.net/sunbaigui/article/details/16843419 最后直接把剩余的链表接过来。
class Solution { public: ListNode *sortList(ListNode *head) { int size = 0; ListNode *node = head; while (node != NULL) { node = node->next; size++; } return mergeSort(head, size); } ListNode* mergeSort(ListNode *node, int size) { if (size == 0 || size == 1) return node; int half = size / 2; ListNode* head1 = node; ListNode* head2 = node; for (int i = 0; i < half; i++) { head2 = head2->next; } head1 = mergeSort(head1, half); head2 = mergeSort(head2, size - half); ListNode *dummy = new ListNode(0); ListNode *prev = dummy; ListNode *n1 = head1; ListNode *n2 = head2; int n1Step = 0; int n2Step = 0; while (n1Step != half && n2Step != size - half) { if (n1->val <= n2->val) { prev->next = n1; n1 = n1->next; n1Step++; } else { prev->next = n2; n2 = n2->next; n2Step++; } prev = prev->next; } while (n2Step != size - half) { prev->next = n2; prev = prev->next; n2 = n2->next; n2Step++; } while (n1Step != half) { prev->next = n1; prev = prev->next; n1 = n1->next; n1Step++; } prev->next = NULL; return dummy->next; } };