代码随想录二刷 |二叉树 | 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;
	}
};

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2023-12-13 20:32:04       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2023-12-13 20:32:04       106 阅读
  3. 在Django里面运行非项目文件

    2023-12-13 20:32:04       87 阅读
  4. Python语言-面向对象

    2023-12-13 20:32:04       96 阅读

热门阅读

  1. linux 内核同步互斥技术之实时互斥锁

    2023-12-13 20:32:04       50 阅读
  2. MYSQL中EXISTS的用法

    2023-12-13 20:32:04       55 阅读
  3. Spring框架中的五种常用设计模式

    2023-12-13 20:32:04       60 阅读
  4. SQL基础理论篇(十二):游标

    2023-12-13 20:32:04       71 阅读
  5. 2023.12面试题汇总小结

    2023-12-13 20:32:04       50 阅读
  6. 【goland如果使用xorm执行mysql的存储过程】

    2023-12-13 20:32:04       48 阅读
  7. python--推导式和生成器表达式

    2023-12-13 20:32:04       61 阅读
  8. 计算机网络技术的应用探讨

    2023-12-13 20:32:04       64 阅读