设为首页 加入收藏

TOP

LeetCode226:Invert Binary Tree
2015-11-21 00:59:01 来源: 作者: 【 】 浏览:1
Tags:LeetCode226:Invert Binary Tree

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
   
     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; } }
   
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
分享到: 
上一篇合并流SequenceInputStream 下一篇HDU 5274(树链剖分)

评论

帐  号: 密码: (新用户注册)
验 证 码:
表  情:
内  容: