C++ 算法:将数组存储的二叉树转换为二叉链表
以下是一个将按顺序方式存储在数组中的二叉树转换为二叉链表形式的算法实现。
#include <iostream>
#include <queue>
using namespace std;
struct TreeNode {
char val;
TreeNode* left;
TreeNode* right;
TreeNode(char x) : val(x), left(NULL), right(NULL) {}
};
// 将按顺序存储的二叉树转换为二叉链表形式
TreeNode* convertToLinkedList(char* c, int n) {
if (n == 0) return NULL;
queue<TreeNode*> q;
TreeNode* root = new TreeNode(c[0]);
q.push(root);
int i = 1;
while (!q.empty() && i < n) {
TreeNode* curr = q.front();
q.pop();
if (c[i] != '.') {
curr->left = new TreeNode(c[i]);
q.push(curr->left);
}
i++;
if (i < n && c[i] != '.') {
curr->right = new TreeNode(c[i]);
q.push(curr->right);
}
i++;
}
return root;
}
// 打印二叉链表形式的二叉树
void printLinkedList(TreeNode* root) {
if (root == NULL) return;
queue<TreeNode*> q;
q.push(root);
while (!q.empty()) {
TreeNode* curr = q.front();
q.pop();
cout << curr->val << ' ';
if (curr->left != NULL) {
q.push(curr->left);
}
if (curr->right != NULL) {
q.push(curr->right);
}
}
cout << endl;
}
int main() {
char c[] = {'A', 'B', 'C', '.', '.', 'D', 'E'};
int n = sizeof(c) / sizeof(c[0]);
TreeNode* root = convertToLinkedList(c, n);
printLinkedList(root);
return 0;
}
运行结果:
A B C D E
该算法首先创建一个队列,用于存储待处理的节点。然后根据数组中的元素创建根节点,并将根节点加入队列中。接下来,从数组的第二个元素开始遍历,每次从队列中取出一个节点,根据数组元素的值创建左子节点和右子节点(如果数组元素不是'.'),并将它们加入队列中。最后,返回根节点,即为二叉链表形式的二叉树。
注意:此算法假设数组中的元素按照二叉树的层次遍历顺序排列。如果数组不满足此条件,可能会得到错误的结果。
原文地址: https://www.cveoy.top/t/topic/qfmf 著作权归作者所有。请勿转载和采集!