1. 程式人生 > >086 Partition List 分隔鏈表

086 Partition List 分隔鏈表

blog pro str SM http ptr nbsp CP desc

給定一個鏈表和一個特定值 x,對鏈表進行分隔,使得所有小於 x 的節點都在大於或等於 x 的節點之前。
你應當保留兩個分區中每個節點的初始相對位置。
例如,
給定1->4->3->2->5->2 和 x = 3,
返回1->2->2->4->3->5.

詳見:https://leetcode.com/problems/partition-list/description/

/**
 * 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 *small=new ListNode(-1);
        ListNode *big=new ListNode(-1);
        ListNode *newSmall=small;
        ListNode *newBig=big;
        while(head)
        {
            if(head->val<x)
            {
                small->next=head;
                small=small->next;
            }
            else
            {
                big->next=head;
                big=big->next;
            }
            head=head->next;
        }
        big->next=nullptr;
        small->next=newBig->next;
        return newSmall->next;
    }
};

參考:https://www.cnblogs.com/springfor/p/3862392.html

086 Partition List 分隔鏈表