前置知识&遍历方法汇总移步:上一篇博文
226# 翻转二叉树
给你一棵二叉树的根节点
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 = [] 输出:[]
提示:
- 树中节点数目范围在
[0, 100]
内-100 <= Node.val <= 100
遍历并交换节点的左右孩子
前序遍历、后序遍历和层序遍历均可,中序遍历需小心避免翻转两次
// 递归法(前序遍历)
// O(n) 0ms; O(n) 12.16MB
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;
}
};
// 迭代法(前序遍历)
// O(n) 0ms; O(n) 12.45MB
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (root == nullptr) return root;
stack<TreeNode*> st;
st.push(root);
while (!st.empty()) {
TreeNode* cur = st.top();
swap(cur->left, cur->right);
st.pop();
if (cur->right) st.push(cur->right);
if (cur->left) st.push(cur->left);
}
return root;
}
};
// 层序遍历
// O(n) 0ms; O(n) 12.4MB
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (root == nullptr) return root;
queue<TreeNode*> que;
que.push(root);
while (!que.empty()) {
int size = que.size();
for (int i = 0; i < size; i++