上下翻转二叉树

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

一个列子:在翻转前

    1
   / \
  2   3
 / \
4   5

翻转后:

   4
  / \
 5   2
    / \
   3   1

先从最简单的情况开始分析:如果只有一个节点,那么返回该节点就可以了(似乎太简单了);如果存在三个节点,父节点和左右子节点(如果有右节点,至少会有一个左节点),那么新的树的结构是左节点变为父节点,右节点变为新的父节点的左节点,原来的父节点变为现在的右节点;

更复杂的情况,考虑,如果左子树已经被翻转了,那么现在应该把父节点和右节点放在哪里?有点递归的味道了;从例子里可以看出,节点1和节点3需要作为节点2的子节点;而节点2是左子树翻转后的最右边的节点;所以回答放在哪里的问题,只要找到新的子树的最右的子节点,并把当前父节点和右节点连接上去就可以了;

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;
}


你可能感兴趣的:(java,LeetCode)