LeetCode ---- 86. 分隔链表 C

86. 分隔链表

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

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

输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]
示例 2：

输入：head = [2,1], x = 2
输出：[1,2]

思路：

建立两个新链表，list1 和 list2 。

遍历原链表，将小于目标值的节点插入list1，大于等于目标值的节点放入list2 。

最后连接两个新链表。返回list1头结点。

typedef struct ListNode Node;

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

    Node* p1head = (Node*)malloc(sizeof(Node)); //链表1
    Node* p2head = (Node*)malloc(sizeof(Node)); //链表2

    Node* p1cur = p1head; //保留链表头
    Node* p2cur = p2head;

    Node* p = head; //保留链表头
    
    while(p)
    {
    
    
        if(p->val < x)  
        {
    
    
            p1cur->next = p;
            p1cur = p;
        }
        else
        {
    
    
            p2cur->next = p;
            p2cur = p;
        }

        p = p->next;
    }
    
    p2cur->next = NULL; //注意将链表末尾置为空
    p1cur->next = p2head->next; //连接两个链表

    return p1head->next;
}