LetCode 86. 分隔链表

最新推荐文章于 2024-05-30 10:12:07 发布

原创最新推荐文章于 2024-05-30 10:12:07 发布 · 238 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

127 篇文章

订阅专栏

本文介绍了一种链表分区算法的实现方法，该算法能够将链表中小于特定值x的所有元素移动到大于等于x的元素之前，同时保持原有的相对顺序不变。通过创建辅助节点和双指针技巧，有效地完成了链表的重排。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        if (head == NULL)
            return head;
        // 创建一个头结点
        ListNode* voidptr = new ListNode(-1);
        voidptr->next = head;
        ListNode* curNode = head;
        ListNode* prv_curNode;
        ListNode* lastNode = voidptr;
        // 先找到一个大于等于x的节点
        while(curNode != NULL && curNode->val < x)
            curNode = curNode->next, lastNode = lastNode->next;
        // 这个节点不存在，就直接返回
        if (curNode == NULL)
            return head;
        // 创建2个节点用来控制链表
        prv_curNode = curNode;
        curNode = curNode->next;
        while(curNode != NULL){
            if (curNode->val < x){
                prv_curNode->next = curNode->next;
                curNode->next = lastNode->next;
                lastNode->next = curNode;
                lastNode = curNode;
                curNode = prv_curNode->next;
            }
            else{
                prv_curNode = curNode;
                curNode = curNode->next;
            }
        }
        return voidptr->next;
    }
};

static int x=[](){
    std::ios::sync_with_stdio(false);
    cin.tie(NULL);
    return 0;
}();