欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

113. 路径总和 II

程序员文章站 2022-05-20 11:42:57
...

路径总和 II

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树,以及目标和 sum = 22,

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

返回:

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

思路+代码+注释:

public List<List<Integer>> pathSum(TreeNode root, int sum) {
            /*
            思路:采用递归从根节点开始遍历树的节点,将树的节点添加到out集合中,当为叶子节点时,如果节点的和等于目标和那么将out集合添加到res中,否则不添加
            每次添加完后都将之前添加的节点从out中删除
             */
            List<Integer> out=new ArrayList<>();
            List<List<Integer>> res=new ArrayList<>();
            if (root==null)
            {
                return res;
            }
            pathSum(root,sum,res,out,0);
            return res;
    }

    private void pathSum(TreeNode node,int tSum,List<List<Integer>> res,List<Integer> out,int curSum)
    {
        out.add(node.val);
        if (node.left==null && node.right==null)
        {
            curSum=curSum+node.val;
            if (curSum==tSum)
            {
                res.add(new ArrayList<>(out));
            }
            return;
        }
        if (node.left!=null)
        {
            pathSum(node.left,tSum,res,out,curSum+node.val);
            out.remove(out.size()-1);
        }
        if (node.right!=null)
        {
            pathSum(node.right,tSum,res,out,curSum+node.val);
            out.remove(out.size()-1);
        }
    }
相关标签: 路径总和 II