题目链接
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/flatten-binary-tree-to-linked-list/solution/er-cha-shu-zhan-kai-wei-lian-biao-by-leetcode-solu/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
代码:
关于方法三,这里有一个非常好理解的动画,点击,具体还是看官网动画吧,我好像解释不清楚,不过也不难理解,看动画很好理解。
思路:方法三的思路非常巧妙,我刚开始想用递归实现这一过程。但是没有处理好。
方法一,递归先序遍历(省略)
方法二,栈实现先序遍历(省略)
方法三,直接左子树链接上右子树,这里要考虑,左右子树有可能为空的情况,以及还有分支的情况。具体实现思路看代码
class Solution {
public void flatten(TreeNode root) {
TreeNode curr = root;
while (curr != null) {
//如果左子树不为空
if (curr.left != null) {
//拿出左子树
TreeNode next = curr.left;
//指向左子树的节点,因为不能直接操作左子树,不然左子树会乱
TreeNode predecessor = next;
//处理左子树的右子树,找到左子树的右子树的右子树的最后一个节点
while (predecessor.right != null) {//
predecessor = predecessor.right;
}
//让左子树的右子树的最后一个节点指向原来根的右子树
predecessor.right = curr.right;
//将原来左子树置为null
curr.left = null;
//左子树链接上右子树
curr.right = next;
}
curr = curr.right;
}
}
}