【劍指offer】34.二叉樹中和爲某一值的路徑

34.二叉樹中和爲某一值的路徑

面試題34. 二叉樹中和爲某一值的路徑

難度中等27

輸入一棵二叉樹和一個整數，打印出二叉樹中節點值的和爲輸入整數的所有路徑。從樹的根節點開始往下一直到葉節點所經過的節點形成一條路徑。

示例:

給定如下二叉樹，以及目標和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

[
   [5,4,11,2],
   [5,8,4,5]
]

//回溯
// 遞歸解決 終止條件節點爲null 返回
// 當target=0 並且左右子節點都爲null的時候 進行數據添加。
// 否則的話 直接遞歸左右子節點。
List<List> ret = new ArrayList<>();

public List<List<Integer>> pathSum(TreeNode root, int sum) {
    pathSum(root,sum,new ArrayList());
    return ret;
}

public void pathSum(TreeNode root,int target,List<Integer> path){
    if(root == null){
        return;
    }
    path.add(root.val);
    target -= root.val;
    if(target == 0 && root.left == null && root.right == null){
        ret.add(new ArrayList<>(path));
    }else{
        pathSum(root.left,target,path);
        pathSum(root.right,target,path);
    }
    //回溯，狀態重置，只添加不刪除就是錯的路徑了，因爲整個遞歸過程用的是同一個收集器path
    path.remove(path.size()-1);
}