package tree;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Stack;

/**
 * 113. 路徑總和 II
 * 給你二叉樹的根節點 root 和一個整數目標和 targetSum ，找出所有 從根節點到葉子節點 路徑總和等於給定目標和的路徑。
 *
 * 葉子節點 是指沒有子節點的節點。
 *
 * @author Tang
 * @date 2021/7/16
 */
public class PathSum {

    int targetSum;

    List<List<Integer>> lists = new ArrayList<>();

    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        if(root == null){
            return lists;
        }
        this.targetSum = targetSum;
        preSearch(root, 0, new Stack<>());
        return lists;
    }

    /**
     * 繼續前序遍歷
     */
    private void preSearch(TreeNode node, int sum, Stack<Integer> stack){
        if(node == null){
            return;
        }
        sum += node.val;
        stack.push(node.val);
        if(node.left == null && node.right == null){
            if(sum == targetSum){
                List<Integer> list = new ArrayList<>(stack);
                lists.addAll(Collections.singleton(list));
                return;
            }
        }
        if(node.left != null){
            preSearch(node.left, sum, stack);
            stack.pop();
        }
        if(node.right != null){
            preSearch(node.right, sum, stack);
            stack.pop();
        }
    }

    public static void main(String[] args) {

    }

}