Partition List 划分链表

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.

1、最容易想到的方法就是分别建立两个链表,一个放大于给定值的,一个放小于给定值的,最后合并,但按这种思路实现最后超时了。

    

2、参考https://www.cnblogs.com/grandyang/p/4321292.html方法二,稍微改了一下思路就能通过:

     将小于给定值的部分组成一个链表,剩余部分组成一个链表,不同的地方在于,剩余部分直接在链表上修改,而不是重新创建链表。

     

class Solution {
public:
    ListNode* partition(ListNode* head, int x) {
        
        if(!head) return head;
        
        ListNode* dummy = new ListNode(-1);
        dummy->next = head;
        
        ListNode* newdummy = new ListNode(-1);
        
        ListNode* cur = dummy, *p = newdummy;
        
        while(cur->next)
        {
            if(cur->next->val < x)
            {
                p->next = cur->next;
                p = p->next;
                cur->next = cur->next->next;
                p->next = nullptr;
            }
            else
            {
                cur = cur->next;
            }
        }
        
        p->next = dummy->next;
        return newdummy->next;
    }
};

猜你喜欢

转载自www.cnblogs.com/573177885qq/p/9613650.html