程序员

86. Partition List

2017-11-09  本文已影响0人  Nautilus1

题目描述:给一个链表和一个值 x ,使得所有比 x 小的值都在比 x 大的值的结点前面,保持两部分的原始相对位置。如:

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

分析:设两指针分别链接比 x 大的结点和比 x 小的结点,最后用链接小于 x 结点的链表链接大于 x 结点的链表即可。时间复杂度O(n),空间O(1)。

代码

/**
 * 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 l(-1);
        ListNode r(-1);
        auto lc = &l;        //指向小于 x 的链表尾
        auto rc = &r;
        for (ListNode *cur = head; cur; cur = cur->next)
        {
            if (cur->val < x)
            {
                lc->next = cur;
                lc = cur;
            }
            else
            {
                rc->next = cur;
                rc = cur;
            }
        }
        lc->next = r.next;
        rc->next = nullptr;
        return l.next;
    }
};
上一篇下一篇

猜你喜欢

热点阅读