在 C++ 中构建一个来自 2D 矩阵的链表


假设我们有一个矩阵,我们需要使用递归方法将其转换为二维链表。

该链表将具有右指针和下指针。

因此,如果输入如下所示:

102030
405060
708090

则输出将是

要解决此问题,我们将遵循以下步骤:

  • 定义函数 make_2d_list(),它将采用矩阵 mat、i、j、m、n,

  • 如果 i 和 j 不是矩阵边界,则:

    • 返回 null

  • temp := 创建一个值为 mat[i, j] 的新节点

  • temp 的右侧 := make_2d_list(mat, i, j + 1, m, n)

  • temp 的下方 := make_2d_list(mat, i + 1, j, m, n)

  • 返回 temp

示例

让我们看看以下实现以获得更好的理解:

 实时演示

#include <bits/stdc++.h>
using namespace std;
class TreeNode {
   public:
   int data;
   TreeNode *right, *down;
   TreeNode(int d){
      data = d;
      right = down = NULL;
   }
};
void show_2d_list(TreeNode* head) {
   TreeNode *right_ptr, *down_ptr = head;
   while (down_ptr) {
      right_ptr = down_ptr;
      while (right_ptr) {
         cout << right_ptr->data << " ";
         right_ptr = right_ptr->right;
      }
      cout << endl;
      down_ptr = down_ptr->down;
   }
}
TreeNode* make_2d_list(int mat[][3], int i, int j, int m, int n) {
   if (i > n - 1 || j > m - 1)
      return NULL;
   TreeNode* temp = new TreeNode(mat[i][j]);
   temp->right = make_2d_list(mat, i, j + 1, m, n);
   temp->down = make_2d_list(mat, i + 1, j, m, n);
   return temp;
}
int main() {
   int m = 3, n = 3;
   int mat[][3] = {
      { 10, 20, 30 },
      { 40, 50, 60 },
      { 70, 80, 90 } };
   TreeNode* head = make_2d_list(mat, 0, 0, m, n);
   show_2d_list(head);
}

输入

{ { 10, 20, 30 },
{ 40, 50, 60 },
{ 70, 80, 90 } }

输出

10 20 30
40 50 60
70 80 90

更新于:2020 年 8 月 27 日

896 次浏览

开启您的 职业

完成课程获得认证

开始
广告