码迷,mamicode.com
首页 > 其他好文 > 详细

上下翻转二叉树

时间:2015-08-15 12:10:33      阅读:196      评论:0      收藏:0      [点我收藏+]

标签:

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


上下翻转二叉树

标签:

原文地址:http://my.oschina.net/u/922297/blog/492847

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!