单链表经典算法题之分割链表

给定一个头结点和一个值x，是链表中所有小于x的值都在x前面

 typedef struct ListNode ListNode;

struct ListNode* partition(struct ListNode* head, int x) {

    //思路一：在原链表上进行修改

    //思路二：创建新链表，使用哨兵位，比x大的尾插，比x小的头插

    //思路三：创建两个链表，一个是大链表，一个是小链表，都整一个哨兵位

    if(head == NULL)

    {

        return head;

    }

    ListNode* lesshead = (ListNode*)malloc(sizeof(ListNode));

    ListNode* greaterhead = (ListNode*)malloc(sizeof(ListNode));

    ListNode* lesstail = lesshead;

    ListNode* greatertail = greaterhead;

    ListNode* pcur = head;

    while(pcur)

    {

        if(pcur->val < x)

        {

            lesstail->next = pcur;

            lesstail = lesstail->next;

        }

        else

        {

            greatertail->next = pcur;

            greatertail = greatertail->next;

        }

        pcur = pcur->next;

    }

    //小链表的尾节点与大链表的第一个有效节点结合

    greatertail->next = NULL;//防止死循环，并将next指针初始化

    lesstail->next = greaterhead->next;

    ListNode* ret = lesshead->next;

   

    free(lesshead);

    free(greaterhead);

    lesshead = greaterhead = NULL;

    return ret;

}

//超出时间限制只有一种情况：就是代码出现了死循环

//创建新链表时，若进行尾插，则要考虑