LeetCode226:Invert Binary Tree

来源:互联网 发布:mac 桌面显示便签 编辑:程序博客网 时间:2024/05/22 10:23

nvert a binary tree.

这里写图片描述
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;        }       }
0 0
原创粉丝点击