代码随想录算法训练营第十七天|LeetCode110,257,404
110.平衡二叉树
平衡二叉树:任意节点的左右子树的高度差不大于1,既然是高度了,那么考虑后续遍历
递归法:
- 明确递归函数的参数和返回值
参数:当前传入节点。
返回值:以当前传入节点为根节点的树的高度。
那么如何标记左右子树是否差值大于1呢?
如果当前传入节点为根节点的二叉树已经不是二叉平衡树了,还返回高度的话就没有意义了。
所以如果已经不是二叉平衡树了,可以返回-1 来标记已经不符合平衡树的规则了。 - 明确终止条件
递归的过程中依然是遇到空节点了为终止,返回0,表示当前节点为根节点的树高度为0 - 明确单层递归的逻辑
如何判断以当前传入节点为根节点的二叉树是否是平衡二叉树呢?当然是其左子树高度和其右子树高度的差值。
分别求出其左右子树的高度,然后如果差值小于等于1,则返回当前二叉树的高度,否则返回-1,表示已经不是二叉平衡树了。
public boolean isBalanced(TreeNode root) {
int result = getHeight(root);
return result==-1?false:true;
}
public int getHeight(TreeNode node){
if (node == null) return 0;
int left = getHeight(node.left);
if (left==-1){
return -1;
}
int right = getHeight(node.right);
//如果右节点的子树已经不为平衡二叉树了,直接返回
if (right == -1){
return -1;
}
//如果左右子树均为平衡二叉树,但是高度差大于1了,直接返回-1
if(Math.abs(left-right)>1) return -1;
return Math.max(left,right)+1;
}
257. 二叉树的所有路径
这道题目要求从根节点到叶子的路径,所以需要前序遍历,这样才方便让父节点指向孩子节点,找到对应的路径。
在这道题目中将第一次涉及到回溯,因为我们要把路径记录下来,需要回溯来回退一个路径再进入另一个路径。
1.递归函数函数参数以及返回值
要传入根节点,记录每一条路径的path,和存放结果集的result,这里递归不需要返回值
2.确定终止条件:
本题可以在找到叶子条件就结束递归。
3.确定单层递归逻辑
因为是前序遍历,需要先处理中间节点,中间节点就是我们要记录路径上的节点,先放进path中。
然后是递归和回溯的过程,在这里递归的时候,如果为空就不进行下一层递归了,因为path 不能一直加入节点,它还要删节点,然后才能加入新的节点,所以需要进行回溯。回溯和递归是一一对应的,有一个递归,就要有一个回溯,通过回溯对已经处理的节点进行删除。
public List<String> binaryTreePaths(TreeNode root) {
List<String>result = new ArrayList<>();
List<Integer> path = new ArrayList<>();
if (root == null){
return result;
}
traversal(root,path,result);
return result;
}
public void traversal(TreeNode node, List<Integer> path, List<String> result){
//对中间节点的处理放到这,要不然后面的叶子节点的值就加不到路径中了
path.add(node.val);
//终止条件,找到叶子节点就可以结束了。
if (node.left == null && node.right == null){
//将path处理成题目要求的样子;
StringBuilder sb = new StringBuilder();
for (int i = 0; i < path.size() - 1;i++){
sb.append(path.get(i)).append("->");
}
sb.append(path.get(path.size()-1));
result.add(sb.toString());
return;
}
//左
if (node.left!=null){
traversal(node.left,path,result);
path.remove(path.size()-1);//回溯,把已经处理了的节点弹出
}
if (node.right!=null){
traversal(node.right, path, result);
path.remove(path.size()-1);
}
}
404.左叶子之和
递归三部曲:
- 确定递归函数的参数和返回值
判断一个树的左叶子节点之和,那么一定要传入树的根节点,递归函数的返回值为数值之和,所以为int
- 确定终止条件
如果遍历到空节点,那么左叶子值一定是0
- 确定单层递归逻辑
当遇到左叶子节点的时候,记录数值,然后通过递归求取左子树左叶子之和,和 右子树左叶子之和,相加便是整个树的左叶子之和。
public int sumOfLeftLeaves(TreeNode root) {
if (root == null) return 0;
//找到叶子节点可以提前返回
if (root.left == null&& root.right == null) return 0;
//获取左子树的左节点之和
int leftSum = sumOfLeftLeaves(root.left);
if (root.left!=null && root.left.left == null&& root.left.right==null){
leftSum = root.left.val;
}
int rightSum = sumOfLeftLeaves(root.right);
return leftSum + rightSum;
}