大家好,我是开车的阿Q,自动驾驶的时代已经到来,没时间解释了,快和阿Q一起上车。作为自动驾驶系统工程师,必须要有最好的C++基础,让我们来一起刷题吧。
题目考察的知识点
这道题目考察了二叉树的展开,需要按照先序遍历的顺序将二叉树展开成一个单链表。
题目解答方法的文字分析
我们可以使用递归来解决这个问题。对于每一个节点,我们将它的左子树展开成一个链表,然后将它的右子树展开成一个链表,最后将左子树链表接在根节点的右子树上,然后将右子树链表接在左子树链表的末尾。这样就完成了二叉树的展开。
举个例子来帮助理解:
假设有以下二叉树:
1
/ \
2 5
/ \ \
3 4 6
展开后的单链表应该是:1 -> 2 -> 3 -> 4 -> 5 -> 6
本题解析所用的编程语言
本题解析所用的编程语言是 C++。
完整且正确的编程代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
TreeNode* flattenTree(TreeNode* root) {
if (!root) {
return nullptr; // 如果当前节点为空,直接返回nullptr
}
TreeNode* leftFlat = flattenTree(root->left); // 递归展开左子树
TreeNode* rightFlat = flattenTree(root->right); // 递归展开右子树
if (leftFlat) {
root->right = leftFlat; // 将左子树链表接在根节点的右子树上
root->left = nullptr; // 清空左子树
while (leftFlat->right) {
leftFlat = leftFlat->right; // 找到左子树链表的末尾节点
}
leftFlat->right = rightFlat; // 将右子树链表接在左子树链表的末尾
} else {
root->right = rightFlat; // 如果左子树为空,直接将右子树链表接在根节点的右子树上
root->left = nullptr; // 清空左子树
}
return root;
}
};
在这段代码中,我们先递归展开左子树和右子树,然后根据左子树链表是否为空来连接节点。如果左子树链表不为空,我们将它连接到根节点的右子树上,并将右子树链表接在左子树链表的末尾。如果左子树链表为空,直接将右子树链表接在根节点的右子树上。

京公网安备 11010502036488号