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, ]
廣告
資料結構
網路
關係型資料庫管理系統
作業系統
Java
iOS
HTML
CSS
Android
Python
C 語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP