86. Partition List

最新推荐文章于 2024-05-27 17:18:09 发布

原创最新推荐文章于 2024-05-27 17:18:09 发布 · 209 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种链表分区算法的实现方法，该算法能够确保所有小于给定值 x 的节点出现在大于等于 x 的节点之前，同时保持各分区内部节点原有的相对顺序。通过具体的代码示例展示了如何使用两个辅助链表来分别收集小于 x 和大于等于 x 的节点，最后将二者连接起来形成最终的分区链表。

Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.

You should preserve the original relative order of the nodes in each of the two partitions.

For example,
Given 1->4->3->2->5->2 and x = 3,
return 1->2->2->4->3->5.

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        ListNode left(0), right(0);
        ListNode *l = &left, *r = &right;

        while(head){
            ListNode* & ref = head->val < x ? l : r;
            ref->next = head;
            ref = ref->next;
            
            head = head->next;
        }
        l->next = right.next;
        r->next = NULL;
        return left.next;
    }
};