使用 C++ 展平链表


在这个问题中,我们提供了一个由两个指针节点组成的链表,即右和下。

  • 右节点是主链表指针。

  • 下节点用于以该节点开头的辅助链表。

所有链表都已排序。

我们的任务是创建一个程序来展平链表,结果列表本身也是一个已排序的列表。

让我们举个例子来理解这个问题

输入

输出

1-> 9-> 8 -> 4 -> 6-> 7-> 2-> 3-> 5

解决方案方法

针对这个问题的一个解决方案是使用链表归并排序。此方法将按照排序顺序递归合并列表,以形成一个展平列表。

示例

展示我们的解决方案如何工作的程序

#include <bits/stdc++.h>
using namespace std;

class Node{
   public:
   int data;
   Node *right, *down;
};
Node* head = NULL;
Node* mergeList(Node* a, Node* b){
   if (a == NULL)
      return b;
   if (b == NULL)
      return a;
   Node* result;
   if (a->data < b->data){
      result = a;
      result->down = mergeList(a->down, b);
   }
   else{
      result = b;
      result->down = mergeList(a, b->down);
   }
   result->right = NULL;
   return result;
}
Node* flattenLinkedList(Node* root){
   if (root == NULL || root->right == NULL)
      return root;
   root->right = flattenLinkedList(root->right);
   root = mergeList(root, root->right);
   return root;
}
Node* push(Node* head_ref, int data){
   Node* new_node = new Node();
   new_node->data = data;
   new_node->right = NULL;
   new_node->down = head_ref;
   head_ref = new_node;
   return head_ref;
}
int main(){
   head = push(head, 7);
   head = push(head, 1);
   head->right = push(head->right, 11);
   head->right = push(head->right, 5);
   head->right = push(head->right, 4);
   head->right->right = push(head->right->right, 12);
   head->right->right = push(head->right->right, 6);
   head->right->right->right = push(head->right->right->right, 8);
   head->right->right->right->right = push(head->right->right->right->right, 16);
   head = flattenLinkedList(head);
   cout<<"The Flattened Linked list is : \n";
   Node* temp = head;
   while (temp != NULL){
      cout<<temp->data<<" => ";
      temp = temp->down;
   }
   cout<<"NULL";
   return 0;
}

输出

The Flattened Linked list is :
1 => 4 => 5 => 6 => 7 => 8 => 11 => 12 => 16 => NULL

更新于:2022 年 2 月 1 日

378 次浏览

开启你的职业生涯

通过完成课程获得认证

开始
广告