给你二叉树的根结点 root
,请你将它展开为一个单链表:
- 展开后的单链表应该同样使用
TreeNode
,其中right
子指针指向链表中下一个结点,而左子指针始终为null
。 - 展开后的单链表应该与二叉树 先序遍历 顺序相同。
示例 1:
输入:root = [1,2,5,3,4,null,6] 输出:[1,null,2,null,3,null,4,null,5,null,6]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [0] 输出:[0]
提示:
- 树中结点数在范围
[0, 2000]
内 -100 <= Node.val <= 100
进阶:你可以使用原地算法(O(1)
额外空间)展开这棵树吗?
思路:我们知道前序遍历的访问顺序为根->左->右,可以观察到在遍历时二叉树最左边的顺序就是访问顺序,只有当前访问的节点没有左节点时,才会开始访问右节点,所以,我们可以用pre记录前一个访问的节点, 站在pre节点的角度来看,
若pre节点有左孩子,下一个访问的即是pre的左孩子
若是pre没有左孩子,则下一个访问pre右分支上的节点Node,且可以将pre的左分支指向下一个访问的节点Node,实现将访问顺序记录到二叉树的左分支上,从根节点开始访问其左孩子,可以得到先序遍历的顺序
由于题目要求存储在右孩子,因此遍历树的左孩子,将当前节点的左孩子节点移到右孩子上,并将左孩子置为空。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* pre=nullptr;
void flat(TreeNode* root){
if(root==NULL)return ;
if(pre==nullptr || pre->left!=NULL) ;
else if(pre->left==NULL){
pre->left=root;
}
pre=root;
flat(root->left);
flat(root->right);
}
void flatten(TreeNode* root) {
TreeNode* temp=root;
flat(temp);
//全部换到右边
TreeNode* p=root;
while(p!=NULL &&p->left!=NULL){
p->right=p->left;
p->left=NULL;
p=p->right;
}
}
};
错解1:错因:没有在将左子树上节点移到右子树后,将左子树上节点置为NULL
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* pre=nullptr;
void flat(TreeNode* root){
if(root==NULL)return ;
if(pre==nullptr || pre->left!=NULL) ;
else if(pre->left==NULL){
pre->left=root;
}
pre=root;
flat(root->left);
flat(root->right);
}
void flatten(TreeNode* root) {
TreeNode* temp=root;
flat(temp);
//全部换到右边
TreeNode* p=root;
while(p!=NULL &&p->left!=NULL){
p->right=p->left;
p=p->right;
}
}
};