86. 分隔链表

题目描述

点这里

思路分析

链表模拟题。
设两个虚拟头节点和对应的尾节点。遍历原链表的所有节点，判断是否有$val<x$，然后涉及链表节点的修改。最后连接两个链表的首尾以及最后指向$NULL$.

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        auto left=new ListNode(-1),right=new ListNode(-1);
        auto left_end=left,right_end=right;
        for(auto p=head;p;p=p->next){
            if(p->val<x) left_end=left_end->next=p;
            else right_end=right_end->next=p;
        }
        left_end->next=right->next;
        right_end->next=NULL;
        return left->next;
    }
};