题解 | #二叉树中和为某一值的路径(二)#

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

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

import java.util.ArrayList;
import java.util.Stack;

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/


public class Solution {
    private ArrayList<ArrayList<Integer>> paths = new ArrayList<>();
    private Stack<Integer>path = new Stack<>();//栈 
    
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int expectNumber) {
        //遍历 所有可能的点都存到list中 最后走一遍
        if(root==null) return paths;
        path.push(root.val);
        expectNumber -= root.val;
        
        if(expectNumber==0 && root.right==null && root.left==null){
            paths.add(new ArrayList<Integer>(path));
        }
        FindPath(root.left,expectNumber);
        FindPath(root.right,expectNumber);
        path.pop();//深搜后回溯
        return paths;
    }
}

ArrayList中有一个 构造,可以传入 Collection接口的实现类。

因此直接用stack记录点,此后直接生成ArrayList

思路:dfs遍历,注意回溯

path.pop();:深搜后回溯

全部评论

相关推荐

11-15 19:28
已编辑
蚌埠坦克学院 硬件开发
点赞 评论 收藏
分享
一颗宏心:华为HR晚上过了十二点后还给我法消息。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务