leetcode刷题/二叉树 226. 翻转二叉树

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;
    }
};

总结:

这道题只要能想到翻转其实就是全部节点都交换一次,就能完成.主要是注意不能使用中序,其中我觉得最容易写的就是层序.所以用了层序写.