代码随想录二刷 |二叉树 | 226.翻转二叉树

题目描述

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

解题思路

其实就是交换一下每一个节点的左右孩子。

这道题目使用前序、后序都可以,中序会将一些节点翻转两次。

这里使用前序遍历。

递归法

递归三部曲:

  1. 确定递归函数的参数和返回值
    参数就是传入节点的指针,题目要求返回其根节点,因此可以用其提供的TreeNode*

    TreeNode* invertTree(TreeNode root)
    
  2. 确定递归函数的终止条件
    当节点为空,递归终止。

    if (root == NULL) return root;
    
  3. 确定递归的逻辑
    前序遍历:中左右。先交换左右孩子节点,然后反转左子树,反转右子树。

    swap(root.left, root.right);
    invertTree(root->left);
    invertTree(root->right);
    

迭代法

深度优先遍历

模板题,注意交换逻辑即可。

广度优先遍历(层次遍历)

模板题,注意交换逻辑即可。

代码实现

递归法

// 前序遍历
class Solution {
public:
	TreeNode* invertTree(TreeNode* root) {
		if (root == NULL) return root;
		swap(root.left, root.right);
		invertTree(root->left);
		invertTree(root->right);
		return root;
	}
};

迭代法

深度优先遍历

// 前序遍历:中左右
// 入栈顺序:右左中
class Solution {
public:
	TreeNode* invertTree(TreeNode* root) {
		stack<TreeNode*> st;
		if (root != NULL) st.push(root);
		while (!st.empty()) {
			TreeNode* node = st.top();
			if (node != NULL) {
				st.pop();
				if (node->right) st.push(node->right);
				if (node->left) st.push(node->left);
				st.push(node);
				st.push(NULL);
			} else {
				st.pop();
				node = st.top();
				st.pop();
				swap(node->left, node->right);
			}
		}
		return root;
	}
};

广度优先遍历(层次遍历)

// 前序遍历:中左右
class Solution {
public:
	TreeNode* invertTree(TreeNode* root) {
		queue<TreeNode*> que;
		if (root !+ NULL) que.push(root);
		while (!que.empty()) {
			int size = que.size();
			for (int i = 0; i < size; i++) {
				TreeNode* 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;
	}
};
  • 9
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值