Hello, 大家好,今天是我参加 9 月更文的第 8 天,今天给大家带来的关于二叉树相关的算法题是二叉树的左叶子之和,正文如下:
题目
翻转一棵二叉树。
示例:
输入:
输出:
解题思路
根据题意,翻转二叉树的意思就是把每个节点的左右孩子翻转,这样就可以达到整体的翻转效果了。
所以,我们可以利用先序遍历,中序遍历,后续遍历这三种方法中的一种来实现翻转,思路如下:
递归三部曲:
确定递归函数的参数和返回值:递归函数的参数为输入的节点,由于题目中需要返回 TreeNode 类型的返回值,所以递归函数的返回值为 TreeNode;
确定终止条件:当 root == null 时,终止递归,返回 root;
确定单层递归逻辑: 如果是先序遍历,则先交换左右孩子节点,然后翻转左子树,再翻转右子树;如果是中序遍历,则先翻转左子树,再交换左右孩子节点,最后翻转右子树;如果是后序遍历,则先翻转左子树,再翻转右子树,最后交换左右孩子节点;
代码实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
// 利用前序遍历
class Solution {
// 先序遍历--从顶向下交换
public TreeNode invertTree(TreeNode root) {
if (root == null) return null;
// 保存右子树
TreeNode rightTree = root.right;
// 交换左右子树的位置
root.right = invertTree(root.left);
root.left = invertTree(rightTree);
return root;
}
}
// 利用中序遍历
class Solution {
public TreeNode invertTree(TreeNode root) {
if (root == null) return null;
invertTree(root.left); // 递归找到左节点
TreeNode rightNode= root.right; // 保存右节点
root.right = root.left;
root.left = rightNode;
// 递归找到右节点 继续交换 : 因为此时左右节点已经交换了,所以此时的右节点为root.left
invertTree(root.left);
}
}
// 利用后序遍历
class Solution {
public TreeNode invertTree(TreeNode root) {
// 后序遍历
if (root == null) return null;
TreeNode leftNode = invertTree(root.left);
TreeNode rightNode = invertTree(root.right);
root.right = leftNode;
root.left = rightNode;
return root;
}
}
复制代码
最后
复杂度分析:
评论