二叉樹翻轉(LeetCode平臺)
阿新 • • 發佈:2019-01-28
二叉樹翻轉
翻轉一棵二叉樹。
示例:
輸入:
4
/ \
2 7
/ \ / \
1 3 6 9
輸出:
4
/ \
7 2
/ \ / \
9 6 3 1
備註:
這個問題是受到 Max Howell 的 原問題 啟發的 :
谷歌:我們90%的工程師使用您編寫的軟體(Homebrew),但是您卻無法在面試時在白板上寫出翻轉二叉樹這道題,這太糟糕了。
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:
void exchange(TreeNode* root){
TreeNode* node=root;
if(node!=NULL){
TreeNode* temp=node->left;
node->left=node->right;
node->right=temp;
}
}
TreeNode* invertTree(TreeNode* root) {
TreeNode* node=root;
if(root==NULL) {
return root;
}
invertTree(node->left);//翻轉左子樹
invertTree(node->right);//翻轉右子樹
exchange(node);//交換左子節點與右子節點
return root;
}
};