226. 翻转二叉树
题意:
翻转一棵二叉树。
示例:
输入:
4
/ \
2 7
/ \ / \
1 3 6 9
输出:
4
/ \
7 2
/ \ / \
9 6 3 1
解题思路:
这道题只需要遍历全部的节点然后交换它的两个孩子节点即可.我使用的是层序遍历,套入模板即可.
唯一注意的就是不能使用中序,因为根据左中右的顺序,左边先交换,然后中间节点交换后,左节点变成右节点,这样就等于左节点交换了两次,右节点没有交换.
迭代法层序遍历模板
queue<TreeNode *> que; //用辅助队列
que.push(root); //先压入栈
TreeNode *node; //用来存储队列头节点
while (!que.empty())
{
node = que.front();
que.pop(); //出队
if (node->left)
{
que.push(node->left);
...
}
if (node->right)
{
que.push(node->right);
...
}
}
return ;
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (!root)
return root;
queue<TreeNode *> que;
que.push(root);
TreeNode *node;
while (!que.empty())
{
node = que.front();
que.pop();
swap(node->left, node->right);
if (node->left)
que.push(node->left);
if (node->right)
que.push(node->right);
}
return root;
}
};
总结:
这道题只要能想到翻转其实就是全部节点都交换一次,就能完成.主要是注意不能使用中序,其中我觉得最容易写的就是层序.所以用了层序写.