C++ 中的分区链表
假如我们有一个链表和一个值 x。我们必须进行分段。这样分段,所有小于 x 的节点都在大于或等于 x 的节点之前。我们应当保留这两个分区中每个节点的原始相对顺序。因此,如果链表形如 [1,4,3,2,5,2],并且 x = 3,那么输出应为 [1,2,2,4,3,5]
为解决此问题,我们将遵循以下步骤 −
- 创建虚拟节点 d1 和 d2,并用 -1 初始化它们,创建两个指针 dp1 和 dp2,它们分别指向 d1 和 d2。
- 当 a 不为空时
- 如果 a 的值 < b
- dp1 的下一个值 := 一个包含 a 值的新节点
- dp1 := dp1 的下一个指针
- 否则
- dp2 的下一个值 := 一个包含 a 值的新节点
- dp2 := dp2 的下一个指针
- a := a 的下一个值
- 如果 a 的值 < b
- dp1 的下一个 := d2 的下一个
- 返回 d1 的下一个
示例 (C++)
让我们看看以下实现以获得更好的理解 −
#include <bits/stdc++.h> using namespace std; class ListNode{ public: int val; ListNode *next; ListNode(int data){ val = data; next = NULL; } }; ListNode *make_list(vector<int> v){ ListNode *head = new ListNode(v[0]); for(int i = 1; i<v.size(); i++){ ListNode *ptr = head; while(ptr->next != NULL){ ptr = ptr->next; } ptr->next = new ListNode(v[i]); } return head; } void print_list(ListNode *head){ ListNode *ptr = head; cout << "["; while(ptr->next){ cout << ptr->val << ", "; ptr = ptr->next; } cout << "]" << endl; } class Solution { public: ListNode* partition(ListNode* a, int b) { ListNode* dummy1 = new ListNode(-1); ListNode* dummy2 = new ListNode(-1); ListNode* dummyPtr1 = dummy1; ListNode* dummyPtr2 = dummy2; while(a){ if(a->val < b){ dummyPtr1->next = new ListNode(a->val); dummyPtr1 = dummyPtr1->next; } else{ dummyPtr2->next = new ListNode(a->val); dummyPtr2 = dummyPtr2->next; } a = a->next; } dummyPtr1->next = dummy2->next; return dummy1->next; } }; main(){ Solution ob; vector<int> v = {1,4,6,3,2,5,2,8}; ListNode *head = make_list(v); print_list(ob.partition(head, 3)); }
输入
[1,4,6,3,2,5,2,8] 3
输出
[1, 2, 2, 4, 6, 3, 5, ]
广告