代码随想录算法训练营第十七天|110.平衡二叉树、257. 二叉树的所有路径、404.左叶子之和
110.平衡二叉树
问题简述:判断二叉树是否为平衡二叉树。
思考:判读二叉树遍历方式为左右根,求高度遍历方式也为左右根。
算法思路:如果当前节点为空或者为叶子结点,则返回true;如果不为空或者为叶子结点,则判断左右子树高度差十分大于1,大于一则直接返回fasle,小于一要判断左右子树收否都为平衡二叉树,如果不是则返回fasle。
class Solution {
public boolean isBalanced(TreeNode root) {
if (root == null) return true;
//叶子结点为平衡二叉树
if (root.left == null && root.right == null) return true;
//左右子树高度大于1不为平衡二叉树
if (Math.abs(getdeep(root.left) - getdeep(root.right)) > 1) return false;
//当前节点的左右子树高度小于1,且左右子树均为为平衡二叉树,返回true
if (isBalanced(root.right) && isBalanced(root.left)) return true;
//左右子树不都为平衡二叉树,返回fasle
return false;
}
//求二叉树当前节点高度
public int getdeep(TreeNode root){
if (root == null) return 0;
return Math.max(getdeep(root.left), getdeep(root.right)) + 1;
}
}
257. 二叉树的所有路径
问题简述:得到二叉树的所有路径。
思考:这道题感觉很难理解,花了不少时间。关键在于不理解回溯。然后路径都是根结点到叶子结点的路径,所以要先遍历跟结点,再遍历左右结点。
算法思路:先序遍历每一个结点,将结点放入path,如果遍历到叶子结点时,将路径path存入StringBuilder,再存入res。如果不是叶子结点,则遍历左结点,递归添加左面结点,存入所有路径后进行回溯。再递归添加右面结点,存入所有路径后进行回溯。最后返回。
import java.util.ArrayList;
import java.util.List;
class Solution {
public List<String> binaryTreePaths(TreeNode root) {
//用于存储最终结果
List<String> res = new ArrayList<>();
//用于存储每条路径值的List
List<Integer> path = new ArrayList<>();
traversal(root, path, res);
return res;
}
public void traversal(TreeNode root, List<Integer> path, List<String> res){
if (root == null) return;
//path中加入当前节点
path.add(root.val);
//到达叶子结点
if (root.left == null && root.right == null){
//此时已到达叶子结点,将path的元素传入StringBuilder
StringBuilder sb = new StringBuilder();
//先添加一个元素
sb.append(path.get(0));
//再不断添加间隔和数字
for (int i = 1; i < path.size(); i++) {
sb.append("->").append(path.get(i));
}
//将字符串加入List
res.add(sb.toString());
return;
}
//左面有结点先添加左面结点
if (root.left != null){
//递归添加左面结点添加
traversal(root.left, path, res);
//回溯到添加这个左结点之前(开始一直没想明白为什么回溯只需要删除一个结点,而不是删除当前root后的所有结点:是因为每次递归都会删除当前添加的一个结点,所以这时候path就只剩下多余的一个结点了)
path.remove(path.size() - 1);
}
if (root.right != null){
//递归添加左面结点
traversal(root.right, path, res);
//回溯到添加这个右结点之前
path.remove(path.size() - 1);
}
//遍历完了所有路径
return;
}
}
404.左叶子之和
问题简述:求出所有左叶子结点和。
思考:这道题也是开始少考虑了,看了leetcode的错误用例才明白错在哪。
算法思路:定义一个sum,当遍历空结点时直接返回0;当前结点为叶子结点时也直接返回0;当遍历到左孩子是叶子结点的的结点时,sum添加当前值。再一次遍历左右子树,将所有左叶子找出添加进sum。
根左右遍历:
class Solution {
int sum = 0;
public int sumOfLeftLeaves(TreeNode root) {
if (root == null) return 0;
//当前结点为叶子结点时也直接返回0,因为在遍历这个结点的父结点时就已经将其判断添加sum
if (root.left == null && root.right == null) return 0;
//当遍历到左孩子是叶子结点的的结点时,sum添加当前值
if (root.left != null && root.left.left == null && root.left.right == null) sum += root.left.val;
//遍历左右结点
sumOfLeftLeaves(root.left);
sumOfLeftLeaves(root.right);
return sum;
}
}
算法思路:当遍历空结点时直接返回0;当前结点为叶子结点时也直接返回0;当遍历到左孩子是叶子结点的的结点时,返回左孩子值与右子树值加合。如果左子树不是叶子结点,则返回两个树的加合
后序遍历:
class Solution {
public int sumOfLeftLeaves(TreeNode root) {
if (root == null) return 0;
当前结点为叶子结点时也直接返回0
if (root.left == null && root.right == null) return 0;
//当遍历到左孩子是叶子结点的的结点时,直接返回左孩子值与右子树的加合,且此时要保证左子树存在
if (root.left != null && root.left.left == null && root.left.right == null) return root.left.val + sumOfLeftLeaves(root.right);
//如果左子树不是叶子结点,则返回两个树的加合
return sumOfLeftLeaves(root.left) + sumOfLeftLeaves(root.right);
}
}
感想
感觉第二题有难度,可能自己想不出来,边看代码示例边写的。