LintCode 96. Partition List

本文介绍了一种链表分区算法,该算法类似于快速排序中的分区过程,但应用于链表而非数组。算法将链表分为两部分,一部分包含所有小于给定值x的节点,另一部分包含所有大于或等于x的节点。通过创建两个虚拟头节点简化实现,最后连接两个链表并返回结果。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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.

  • 类似于快速排序的一部分,只不过由数组变成了链表
  • 比数组更简单,直接分成两根链表,一根left记录小于x的点,一根right记录大于等于x的点
  • 注意记把right的尾部设为NULL,不然会行程闭环,TLE
/**
 * Definition of singly-linked-list:
 * class ListNode {
 * public:
 *     int val;
 *     ListNode *next;
 *     ListNode(int val) {
 *        this->val = val;
 *        this->next = NULL;
 *     }
 * }
 */

class Solution {
public:
    /**
     * @param head: The first node of linked list
     * @param x: An integer
     * @return: A ListNode
     */
    ListNode * partition(ListNode * head, int x) {
        ListNode *left_dummy = new ListNode(0);
        ListNode *left_tail = left_dummy;
        ListNode *right_dummy = new ListNode(0);
        ListNode *right_tail = right_dummy;
        
        while (head != NULL) {
            if (head->val < x) {
                left_tail->next = head;
                left_tail = head;
            } else {
                right_tail->next = head;
                right_tail = head;
            }
            head = head->next;
        } 
        left_tail->next = right_dummy->next;
        right_tail->next = NULL;
        return left_dummy->next;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值