JZ34 二叉树中和为某一值的路径(二)(中)

输入一颗二叉树的根节点root和一个整数expectNumber,找出二叉树中结点值的和为expectNumber的所有路径。

1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点

2.叶子节点是指没有子节点的节点

3.路径只能从父节点到子节点,不能从子节点到父节点

4.总节点数目为n

如二叉树root为{10,5,12,4,7},expectNumber为22

则合法路径有[[10,5,7],[10,12]]

数据范围:

树中节点总数在范围 [0, 5000] 内

-1000 <= 节点值 <= 1000

-1000 <= expectNumber <= 1000

示例1

输入:{10,5,12,4,7},22

返回值:[[10,5,7],[10,12]]

说明:返回[[10,12],[10,5,7]]也是对的

示例2

输入:{10,5,12,4,7},15

返回值:[]

示例3

输入:{2,3},0

返回值:[]

示例4

输入:{1,3,4},7

返回值:[]

 解法:

import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }
}
*/
public class Solution {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
//      创建返回结果
        ArrayList<ArrayList<Integer>> result=new ArrayList<ArrayList<Integer>>();
        
//       如果根节点为空直接返回result
        if(root==null)return result;
        
//       调用find方法获取返回结果
        find(result,new ArrayList<Integer>(),root,target);
        return result;
    }
    public void find(ArrayList<ArrayList<Integer>> result,ArrayList<Integer> list,TreeNode root,int target){
//      首先把根节点的值存入list
        list.add(root.val);
        
//       如果返回的根节点的左右子树为空且最后一个节点的值与target值同则存入result并返回
        if(root.left==null&&root.right==null){
            if(target==root.val){
                result.add(list);
            }
//          如果相等直接return回去,而list2中保存了父节点的值继续走44行
            return;
        }
//       创建中间list2用于保存list中的根节点所有值,用于遍历右子树
        ArrayList<Integer> list2=new ArrayList<>();
        list2.addAll(list);
//      如果左边的子树不为空则先遍历左子树
        if(root.left!=null)find(result,list,root.left,target-root.val);
//      如果右边的子树不为空则遍历右子树
        if(root.right!=null)find(result,list2,root.right,target-root.val);
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_47465999/article/details/121520034
今日推荐