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不为null

    • dp2的下一个:=一个值为a的新节点

    • dp2:= dp2的下一个指针

    • dp1的下一个:=值为a的新节点

    • dp1:= dp1的下一个指针

    • 如果a <b

    • 除此以外

    • 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, ]