给你二叉树的根结点 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)
额外空间)展开这棵树吗?
思路:
实际上左子树的最右边的结点在先序遍历中一定是当前结点的左子树中最后一个遍历到的,也就是排在当前结点右子树所有节点的前面,对于每一个结点都做相同的操作,直到遍历完。
具体做法就是,先找到根节点左子树最右边节点,然后把该节点的右子树添加成原来根节点的右子树,然后把根节点的右子树替换成现在根节点的左子树,再把根节点的左子树置为空,接下来将根节点往下移动,也就是root=root->right,直到遍历完。
class Solution {
public:
void flatten(TreeNode* root) {
while(root)
{
TreeNode* move=root->left;
while(move&&move->right)
{
move=move->right;
}
if(move)
{
move->right=root->right;
root->right=root->left;
root->left=nullptr;
}
root=root->right;
}
}
};