在 C++ 中构建一个来自 2D 矩阵的链表
假设我们有一个矩阵,我们需要使用递归方法将其转换为二维链表。
该链表将具有右指针和下指针。
因此,如果输入如下所示:
10 | 20 | 30 |
40 | 50 | 60 |
70 | 80 | 90 |
则输出将是
要解决此问题,我们将遵循以下步骤:
定义函数 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
广告