上下翻转二叉树

from https://leetcode.com/problems/binary-tree-upside-down/; java

一个列子:在翻转前ide

    1
   / \
  2   3
 / \
4   5

翻转后:code

   4
  / \
 5   2
    / \
   3   1

先从最简单的状况开始分析:若是只有一个节点,那么返回该节点就能够了(彷佛太简单了);若是存在三个节点,父节点和左右子节点(若是有右节点,至少会有一个左节点),那么新的树的结构是左节点变为父节点,右节点变为新的父节点的左节点,原来的父节点变为如今的右节点;递归

更复杂的状况,考虑,若是左子树已经被翻转了,那么如今应该把父节点和右节点放在哪里?有点递归的味道了;从例子里能够看出,节点1和节点3须要做为节点2的子节点;而节点2是左子树翻转后的最右边的节点;因此回答放在哪里的问题,只要找到新的子树的最右的子节点,并把当前父节点和右节点链接上去就能够了;leetcode

public TreeNode upsideDownBinaryTree(TreeNode root) {
    if(root == null || (root.left == null && root.right == null)) {
        return root;
    }

    TreeNode leftRoot = upsideDownBinaryTree(root.left);

    TreeNode tmpNode = leftRoot;

    while(tmpNode.right != null) {
        tmpNode = tmpNode.right;
    }

    tmpNode.left = root.right;
    tmpNode.right = root;

    root.right = null;
    root.left = null;
    return leftRoot;
}
相关文章
相关标签/搜索