C++ 围绕给定值对链表进行分区并保持原始顺序


在本教程中给定一个链表,我们需要将所有小于 x 的数字放在列表的开头,其他数字放在后面。我们还需要保留它们与之前相同的顺序。例如

Input : 1->4->3->2->5->2->3,
x = 3
Output: 1->2->2->3->3->4->5

Input : 1->4->2->10
x = 3
Output: 1->2->4->10

Input : 10->4->20->10->3
x = 3
Output: 3->10->4->20->10

要解决此问题,我们现在需要创建三个链表。当我们遇到一个小于 x 的数字时,我们将它插入第一个列表。现在对于等于 x 的值,我们将其放入第二个列表,对于更大的值,我们将其插入第三个列表。最后,我们将这些列表连接起来并打印最终列表。

查找解决方案的方法

在这种方法中,我们将维护三个列表,即 small、equal、large。现在我们保持它们的顺序并将这些列表连接到一个最终列表中,这是我们的答案。

示例

上述方法的 C++ 代码

#include<bits/stdc++.h>
using namespace std;
struct Node{ // structure for our node
    int data;
    struct Node* next;
};
// A utility function to create a new node
Node *newNode(int data){
    struct Node* new_node = new Node;
    new_node->data = data;
    new_node->next = NULL;
    return new_node;
}
struct Node *partition(struct Node *head, int x){
    struct Node *smallhead = NULL, *smalllast = NULL; // we take two pointers for the list                                                     //so that it will help us in concatenation
    struct Node *largelast = NULL, *largehead = NULL;
    struct Node *equalhead = NULL, *equallast = NULL;
    while (head != NULL){ // traversing through the original list
        if (head->data == x){ // for equal to x
            if (equalhead == NULL)
                equalhead = equallast = head;
            else{
                equallast->next = head;
                equallast = equallast->next;
            }
        }
        else if (head->data < x){ // for smaller than x
            if (smallhead == NULL)
                smalllast = smallhead = head;
            else{
               smalllast->next = head;
               smalllast = head;
           }
        }
        else{ // for larger than x
            if (largehead == NULL)
                largelast = largehead = head;
            else{
                largelast->next = head;
                largelast = head;
            }
        }
        head = head->next;
    }
    if (largelast != NULL) // largelast will point to null as it is our last list
        largelast->next = NULL;
   /**********Concatenating the lists**********/
    if (smallhead == NULL){
        if (equalhead == NULL)
           return largehead;
        equallast->next = largehead;
        return equalhead;
    }
    if (equalhead == NULL){
        smalllast->next = largehead;
        return smallhead;
    }
    smalllast->next = equalhead;
    equallast->next = largehead;
    return smallhead;
}
void printList(struct Node *head){ // function for printing our list
    struct Node *temp = head;
    while (temp != NULL){
        printf("%d ", temp->data);
        temp = temp->next;
   }
}
int main(){
    struct Node* head = newNode(10);
    head->next = newNode(4);
    head->next->next = newNode(5);
    head->next->next->next = newNode(30);
    head->next->next->next->next = newNode(2);
    head->next->next->next->next->next = newNode(50);
    int x = 3;
    head = partition(head, x);
    printList(head);
    return 0;
}

输出

2 10 4 5 30

上述代码的解释

在上面描述的方法中,我们将保留三个链表,其内容按顺序排列。这三个链表将分别包含小于、等于和大于 x 的元素。我们的任务现在简化了。我们需要连接这些列表,然后返回头部。

结论

在本教程中,我们解决了围绕给定值对链表进行分区并保持原始顺序的问题。我们还学习了此问题的 C++ 程序以及我们解决此问题的完整方法(常规方法)。我们可以用其他语言(如 C、Java、Python 和其他语言)编写相同的程序。我们希望您觉得本教程有所帮助。

更新于: 2021 年 11 月 25 日

196 次查看

启动您的 职业生涯

通过完成课程获得认证

开始学习
广告