对于二叉树的路径综合问题,随着约束逐渐放宽,解题的难度也逐渐增强。
- 从根节点到叶节点
- 从任意节点到叶节点
- 从任意节点到任意节点
- 不需要遵循从上到下的顺序(前面的都是需要的)
leetcode上面有四道题,从易到难,分别为:
路径总和
路径总和II
路径总和III
路径总和IV (此题需要会员,因此直接看不到)
还有一道和他们也是一脉相承的:
二叉树中的最大路径和
路径总和
解题思路
只需要简单的递归就可以了
class Solution {
public boolean hasPathSum(TreeNode root, int targetSum) {
if(root == null){
return false;
}
if(root.left ==null && root.right == null && root.val == targetSum){
return true;
}
return hasPathSum(root.left,targetSum-root.val)
||hasPathSum(root.right,targetSum-root.val);
}
}
路径总和II
需要加上一个返回链表,在每次到达叶子节点的时候判断能否得到目标值
class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
helper(root,targetSum,new ArrayList<Integer>());
return res;
}
private void helper(TreeNode root,int targetSum,List<Integer> list){
if(root == null){
return ;
}
if(root.left == null && root.right ==null && root.val==targetSum){
list.add(root.val);
res.add(new ArrayList<>(list));
list.remove(list.size()-1);
return ;
}
list.add(root.val);
helper(root.left,targetSum-root.val,list);
helper(root.right,targetSum-root.val,list);
list.remove(list.size()-1);
}
}
路径总和III
此题有两种方法,分别是双重递归和前缀和方法
双重递归时间复杂度在二叉树没退化时是nOlogn,二叉树退化为链表后就是n^2
前缀和方法时间复杂度是On,但是也需要额外的空间复杂度On来维护前缀和
前缀和方法算是一个通用的方法,标记一下,以后重点关注
递归方法
class Solution {
int res = 0;
public int pathSum(TreeNode root, int sum) {
if(root == null){
return 0;
}