[LeetCode] Sort List

There are many merge-sort solutions at the forum, but very few quicksort solutions. So I post my accepted quicksort solution here.

Well, after reading the problem statement, I intuitively select quicksort since it is able to give an in-place solution and thus costs only constant space. Also, it is O(nlogn) in the expected case though it may become O(n^2) in the worst case.

Then I implement my quicksort solution and test it. I then submit it to the online judge. However, the annoying TLE error occurred. I check for the forums and some people suggested to use random pivoting or duplicate skipping. However, implementing random pivoting is a little costly, I lazily tried to skip the duplicates. And it works! So now comes the following solution . Note that each time I choose the first node as the pivot. Moreover, I create a new_head that points to headfor convenience.

Of course, this solution passes the online judge luckily. If the linked list is like: 100000 -> 99999 -> 99998 -> ... -> 1, it will fail since the subproblems only decrease by 1 at each recursion. However, it seems that the LeetCode OJ does not have this kind of test cases.

    void sortListHelper(ListNode* head, ListNode* tail) {
        if (head -> next == tail) return;
        /* Partition the list. */
        ListNode* pre = head;
        ListNode* cur = head -> next;
        ListNode* pivot = cur;
        while (cur -> next && cur -> next != tail) {        
            if (pivot -> val > cur -> next -> val) {
                ListNode* temp = pre -> next;
                pre -> next = cur -> next;
                cur -> next = cur -> next -> next;
                pre -> next -> next = temp;
            }
            else cur = cur -> next;
        }
        sortListHelper(head, pivot);
        /* Here is the trick. */
        while (pivot -> next != tail && pivot -> next -> val == pivot -> val)
            pivot = pivot -> next;
        if (pivot -> next != tail) sortListHelper(pivot, tail);
    } 

    ListNode* sortList(ListNode* head) {
        ListNode* new_head = new ListNode(0);
        new_head -> next = head;
        sortListHelper(new_head, NULL);
        return new_head -> next;
    }