分隔链表

题目

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

来源：力扣（LeetCode）
链接：分隔链表

代码

NOTES
头结点会使链表的增删链接操作会方便很多，省去了头结点的判断。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */

struct ListNode* partition(struct ListNode* head, int x){
    struct ListNode *smallhead,*bighead,*small,*big,*p,*q;
    smallhead=(struct ListNode *)malloc(sizeof(struct ListNode)); //链接小于x的结点的头结点
    smallhead->next=NULL;
    bighead=(struct ListNode *)malloc(sizeof(struct ListNode)); //链接大于等于x的结点的头结点
    bighead->next=NULL;
    p=head;small=smallhead;big=bighead;
    while(p){
        q=p->next;
        if(p->val<x){
            small->next=p;
            small=small->next;
            small->next=NULL;
        }else{
            big->next=p;
            big=big->next;
            big->next=NULL;
        }
        p=q;
    }

    small->next=bighead->next;//small为smllhead链表最后一个结点
    head=smallhead->next;
    free(smallhead);
    free(bighead);
    return head;

}