JZ24 二叉树中和为某一值的路径,dfs

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

http://www.nowcoder.com/questionTerminal/b736e784e3e34731af99065031301bca

dfs1

public class Solution {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        ans=new ArrayList<>();
        if(root==null) return ans;
        ArrayList<Integer> list=new ArrayList<>();
        list.add(root.val);
        dfs(root, target-root.val, list);
        return ans;
    }
    ArrayList<ArrayList<Integer>> ans;
    void dfs(TreeNode root, int target, ArrayList<Integer> list){
        if(root.left==null&&root.right==null){
            if(target==0) ans.add(new ArrayList<Integer>(list));
            return;
        }
        if(root.left!=null){
            list.add(root.left.val);
            dfs(root.left, target-root.left.val, list);
            list.remove(list.size()-1);
        }
        if(root.right!=null){
            list.add(root.right.val);
            dfs(root.right, target-root.right.val, list);
            list.remove(list.size()-1);
        }     
        return;
    }
}

dfs2

public class Solution {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        ans=new ArrayList<>();
        if(root==null) return ans;
        ArrayList<Integer> list=new ArrayList<>();
        //list.add(root.val);
        dfs(root, target, list);
        return ans;
    }
    ArrayList<ArrayList<Integer>> ans;
    void dfs(TreeNode root, int target, ArrayList<Integer> list){
        if(root==null) return;
        target-=root.val;
        list.add(root.val);
        if(root.left==null&&root.right==null){
            if(target==0) ans.add(new ArrayList<Integer>(list));
            list.remove(list.size()-1);
            return;
        }
        dfs(root.left, target, list);
        dfs(root.right, target, list);  
        list.remove(list.size()-1);
        return;
    }
}

这里无法绕开的是,list每一次都需要remove。就像dfs中visited[]先为true后为false一样。
如果不想这么做,解决办法有两个

  1. 每次递归时,new一个ArrayList
  2. 最终确定路径存在的时候才生成list,每一次往list的前面添加元素。
全部评论

相关推荐

mama3925:灵神是天才,路线不适合正常人
点赞 评论 收藏
分享
03-29 17:05
门头沟学院 Java
asdasdasda...:我前段时间找工作焦虑,有几天连续熬夜熬穿了,然后心脏突然不舒服,立马躺床上睡觉了,然后第二天还是不舒服,去看医生说是心率不齐,吓得我后面天天早早睡觉,调养身体,过了好几天才好过来。所以真的,工作这些东西哪有那么重要,最多钱多一点钱少一点,降低物欲。活着才是最重要的,现在想想真的后怕
如何排解工作中的焦虑
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务