题目链接: \sf 题目链接: 题目链接:
图解:
方法一、 D F S \sf 方法一、DFS 方法一、DFS
递归:
让左子树是翻转后的右子树,
右子树是翻转后的左子树。
∴从叶结点开始,自底向上逐步翻转。递归出口是叶结点的下一层节点。
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (!root) return nullptr;
TreeNode* left = invertTree(root->left);
TreeNode* right = invertTree(root->right);
root->left = right;
root->right = left;
return root;
}
};
另一种写法:
class Solution {
public:
void invert(TreeNode* t)
{
if (!t) return;
TreeNode* temp = t->left;
t->left = t->right;
t->right = temp;
invert(t->left);
invert(t->right);
}
TreeNode* invertTree(TreeNode* root) {
if (!root) return nullptr;
invert(root);
return root;
}
};
方法二、 B F S \sf 方法二、BFS 方法二、BFS
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (!root) return nullptr;
std::queue<TreeNode*> q;
q.push(root);
while (!q.empty()) {
auto node = q.front();
q.pop();
swap(node->left, node->right);
if (node->left) q.push(node->left);
if (node->right) q.push(node->right);
}
return root;
}
};
注:
不能只是交换节点的数据值,而是要把整个节点进行交换。