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

LeetCode226:Invert Binary Tree

时间:2015-06-24 00:49:49      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:

nvert a binary tree.

 4

/ \
2 7
/ \ / \
1 3 6 9
to
4
/ \
7 2
/ \ / \
9 6 3 1
Trivia:
This problem was inspired by this original tweet by Max Howell:
Google: 90% of our engineers use the software you wrote (Homebrew), but you can’t invert a binary tree on a whiteboard so fuck off.
Hide Tags Tree

虽然不知道Homebrew为何物,但是google 90%的人都用的产品绝对是高大上的,所以估且把这个trivia当成Max Howell在卖萌吧。


解法一

反转二叉树使用递归实现,可以看出,如果把左子树和右子树都反转了的话,只需要交换它的左右子树节点就可以了。而反转子树和反转它自身是同一个问题,所以可以使用递归实现。
runtime:4ms

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* invertTree(TreeNode* root) {
        if(root==NULL)
            return NULL;
        TreeNode * leftTree=NULL;
        TreeNode * rightTree=NULL;
        if(root->left)
            leftTree=invertTree(root->left);
        if(root->right)
            rightTree=invertTree(root->right);
        root->left=rightTree;
        root->right=leftTree;
        return root;
    }
};

解法二

这道题还可以使用非递归来实现,非递归实现代码可能会多了一点,但是也很好理解。使用一个队列,开始时将根节点加入队列中,然后交换它的左右子节点并将根节点从队列中弹出,如果左右子节点非空,将左右子节点加入队列中,一直处理到队列中没有元素为止。参考的解答:https://leetcode.com/discuss/40567/my-c-codes-recursive-and-nonrecursive
runtime:0ms
从运行时间可以看出研究一下非递归的解法还是很有意义的。

   class Solution {
    public:
        TreeNode* invertTree(TreeNode* root) {
            queue<TreeNode *> tbpNode;
            if(root) tbpNode.push(root);
            TreeNode *curNode, *temp;
            while(!tbpNode.empty())
            {
                curNode = tbpNode.front();
                tbpNode.pop();
                temp = curNode->left;
                curNode->left = curNode->right;
                curNode->right = temp;
                if(curNode->left) tbpNode.push(curNode->left);
                if(curNode->right) tbpNode.push(curNode->right);
            }
            return root;
        }
       }

LeetCode226:Invert Binary Tree

标签:

原文地址:http://blog.csdn.net/u012501459/article/details/46612311

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