题目描述:
给你二叉树的根结点 root ,请你将它展开为一个单链表:
展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
展开后的单链表应该与二叉树 先序遍历 顺序相同。
算法思想:
这道题的函数是 void flatten(TreeNode root) ,我们尝试给出函数定义:给 flatten 函数输入一个节点 root,那么以 root 为根的二叉树就会被拉平为一条链表。
具体流程如下:
- 将 root 的左子树和右子树拉平。
- 将 root 的右子树接到左子树下方,然后将整个左子树作为右子树。
如何把 root 的左右子树拉平?其实很简单,按照 flatten 函数的定义,对 root 的左右子树递归调用 flatten 函数即可:
public void flatten(TreeNode root) {
if(root == null) return;
flatten(root.left); //将左边的拉平
flatten(root.right); //将右边的拉平
/*
后序遍历操作
*/
TreeNode left = root.left;
TreeNode right = root.right;
//将左边子树移到右边
root.left = null;
root.right = left;
//原本的右边子树接在后面
TreeNode p = root;
while(p.right!=null){
p = p.right;
}
p.right = right;
}
flatten函数是怎么把左右子树拉平的?说不清楚,递归的解题就是不要陷入具体的细节,而是相信这个定义,让root去做它该做的事情。
复杂度分析:
后更