给你一棵二叉树的根节点 root
,翻转这棵二叉树,并返回其根节点。
示例 1:
输入:root = [4,2,7,1,3,6,9] 输出:[4,7,2,9,6,3,1]
示例 2:
输入:root = [2,1,3] 输出:[2,3,1]
示例 3:
输入:root = [] 输出:[]
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
// 递归法
// if(root == nullptr) return root;
// swap(root->left, root->right);
// invertTree(root->left);
// invertTree(root->right);
// return root;
// 用栈来模拟递归
stack<TreeNode*> st;
if(root != nullptr) st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
st.pop();
swap(node->left, node->right);
if(node->right) st.push(node->right);
if(node->left) st.push(node->left);
}
return root;
}
};