Leetdode Partition List

最新推荐文章于 2025-12-16 18:10:48 发布

原创最新推荐文章于 2025-12-16 18:10:48 发布 · 175 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #链表

leetcode 同时被 2 个专栏收录

167 篇文章

订阅专栏

水题

112 篇文章

订阅专栏

本文介绍了一种通过简单模拟的方式实现队列中元素按指定值进行分类的方法，确保小于该值的元素位于前面，大于等于该值的元素位于后面。使用了两个辅助链表分别存放小于和大于等于给定值的节点。

题意：将队列按给定值分类，小的在前，大的在后。

思路：简单模拟。

/**
 * 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) return NULL;
        
        ListNode* shead = NULL;
        ListNode* bhead = NULL;
        ListNode* snext = NULL;
        ListNode* bnext = NULL;
        ListNode* next =head;
        while(next) {
            ListNode* now = next;
            next = next->next;
            
            if(now->val < x) {
                if(!shead) {
                    shead = now;
                    snext = now;
                    snext->next = NULL;
                }
                else {
                    snext->next = now;
                    snext = snext->next;
                    snext->next = NULL;
                }
            }
            else {
                if(!bhead) {
                    bhead = now;
                    bnext = now;
                    bnext->next = NULL;
                }
                else {
                    bnext->next = now;
                    bnext = bnext->next;
                    bnext->next = NULL;
                }
            }
        }
        if(!snext) return bhead;
        else {
            snext->next = bhead;
            return shead;
        }
        
    }
};