Homebrew是OS X平臺上的包管理工具,在用Mac的程序員基本都知道這個工具。
HomeBrew的開發者是Max Howell。然而面試谷歌時卻蛋疼了。Max Howell在Twitter發帖:
twitter
可見,會手寫反轉二叉樹多么重要。正好Leetcode上有這個題目,下面進入正題。
二叉樹是數據結構里一個重要的概念。
而反轉二叉樹的基本意思就是下圖這樣。
Invert a binary tree.
4
/ \
2 7
/ \ / \
1 3 6 9
to:
4
/ \
7 2
/ \ / \
9 6 3 1
每一個節點的左右子樹對換,左右子樹的左右節點也需要交換,這種時候很容易想到的就是遞歸的方法。
下面是在Leetcode 通過的C++代碼:
/**
* 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) {
TreeNode* tmp;
if(!root)
return NULL;
if(root->left)
root->left=invertTree(root->left);
if(root->right)
root->right=invertTree(root->right);
tmp=root->left;
root->left=root->right;
root->right=tmp;
return root;
}
};
至于非遞歸的做法也很簡單,借助一個隊列就可以實現,在C++里,直接使用標準庫的queue就可以。
首先取根節點入隊,再出隊,交換它的左右節點,再將左右節點入隊,這樣就可以以層次遍歷的方法,處理每一層的節點。
/**
* 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) {
queue<TreeNode *> node_queue;
if(root == NULL)
return root;
node_queue.push(root);
while(node_queue.size()>0)
{
TreeNode* pFrontNode = node_queue.front();
node_queue.pop();
TreeNode *tmp = pFrontNode->left;
pFrontNode->left = pFrontNode->right;
pFrontNode->right = tmp;
if(pFrontNode->left)
node_queue.push(pFrontNode->left);
if(pFrontNode->right)
node_queue.push(pFrontNode->right);
}
return root;
}
};
最近在看Python,用Python實現也一樣,遞歸解法:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def invertTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
if root is None:
return None
root.left,root.right = self.invertTree(root.right),self.invertTree(root.left)
return root
非遞歸
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def invertTree(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
queue = collections.deque()
if root:
queue.append(root)
while queue:
node = queue.popleft()
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
node.left,node.right = node.right,node.left
return root