Partition List

本文介绍了一种链表分区算法的实现方法,通过使用两个临时头指针和对应的尾指针来构建小于和大于等于给定值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.

用两个临时变量做头指针,然后用两个指向这两个头指针的指针(tail1, tail2)作为当前可以插入的位置,便利整个list,按照题意要求把大小节点分别接入两个tail上,同时更新tail为指向当前插入节点的next域的指针,最后合并两个list返回。

注:用两个dummy和两个pre指针也可以。一般说来链表类问题,用一个额外的dummy或者这种指向指针的指针都会方便解决问题。


---------------------------
/**
 * 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 || head->next == NULL) return head;
        
        ListNode *first = NULL;
        ListNode *second = NULL;
        ListNode **ftail = &first, **stail = &second;
        
        ListNode *cur = head;
        while (cur != NULL) {
            if (cur->val < x) {
                *ftail = cur;
                ftail = &(cur->next);
            } else {
                *stail = cur;
                stail = &(cur->next);
            }
            cur = cur->next;
        }
        
        *ftail = second;
        *stail = NULL;
        return first;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值