leetcode 86. 分隔链表

最新推荐文章于 2024-11-13 16:38:47 发布

笨比master

最新推荐文章于 2024-11-13 16:38:47 发布

阅读量71

点赞数

分类专栏： new leetcode 文章标签：链表

本文链接：https://blog.youkuaiyun.com/m0_37454852/article/details/115310423

版权

new leetcode 专栏收录该内容

84 篇文章

订阅专栏

86. 分隔链表

在这里插入图片描述

/**
 * 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) {
        ListNode* newHeadLess = new ListNode(0, head);
        ListNode* newHeadgreater = new ListNode();
        ListNode *p = newHeadLess, *q = newHeadgreater;
        while(p->next) {
            if(p->next->val >= x) {
                q->next = p->next;
                q = q->next;
                p->next = q->next;
                q->next = nullptr;
            }
            else p = p->next;
        }
        p->next = newHeadgreater->next;
        delete newHeadgreater;
        return newHeadLess->next;
    }
};