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 的下一个值
  • 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, ]

更新于: 28-Apr-2020

498 次浏览

开启你的职业生涯

完成课程获取认证

开始学习
广告