从头做leetcode之leetcode 86 分割链表

最新推荐文章于 2024-12-08 22:18:23 发布

原创最新推荐文章于 2024-12-08 22:18:23 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #单链表

从头做leetcode 专栏收录该内容

94 篇文章

订阅专栏

本文介绍了一种链表分隔算法，该算法能够将链表中小于特定值x的节点与大于或等于x的节点分隔开，同时保持两部分节点的初始相对位置不变。通过对链表进行遍历，并使用两个指针分别跟踪小于x和大于等于x的节点，最后将两部分链表连接起来，实现链表的有效分隔。

86.分隔链表

给定一个链表和一个特定值 x，对链表进行分隔，使得所有小于 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) {
        ListNode *head1 = new ListNode;
        ListNode *head2 = new ListNode;
        ListNode *p = head1,*q = head2;
        ListNode *s = head;
        while(s){
            if(s->val >= x){
                p->next = s;
                p = p->next;
            }
            else if(s->val < x){
                q->next = s;
                q = q->next;
            }
            s = s->next;
        }
        q->next = head1->next;
        p->next = NULL;
        return head2->next;
    }
};