path_sum2

题目描述

Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum.

For example:

Given the below binary tree and sum = 22,

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

return

[

[5,4,11,2],

[5,8,4,5]

]

解题思路

做此题前可以先尝试path_sum,判断二叉树是否存在一条根节点到叶子节点路径使得数值和等于sum,这里则是求出所有的这样的路径集合。

原先的递归关键点是对左右儿子节点递归,左右有一个返回true,当前函数就返回true,现在我们是计算路径,所以每个节点都需要对左右子树遍历,对于符合条件的路径加入到结果结合中。

为了找到一条路径,我定义了一个SuperNode,可以记录其前驱,数据结构如下:

 public class SuperNode {
     int val;
     SuperNode pre;//指向前驱节点
     public SuperNode(int x)
        {
            val = x;
            pre = null;
        }
 }

每次经过一个节点,我们都建立一个supernode sn,标记该节点的val,以及其前驱,如果找到一条符合条件的通路,则把这个sn加入到我们的节点结合List中,最后对list遍历,对其中每个元素sn,进行找前驱的方式,可以得到一条路径,然后放到路径集合中。

详细代码

//path sum2
public List<List<Integer>> pathSum(TreeNode root, int sum)
{
    List<List<Integer>> resultLists = new ArrayList<>();//结果路径集合
    List<SuperNode> superNodes = new ArrayList<>();//supernodes 集合,每个supernode可以生成一条路径。
    if(root == null)
    {
        return resultLists;
    }
    findPathSum(root, sum, new SuperNode(root.val), superNodes);
    for(SuperNode sn:superNodes)//对supernodes进行遍历,每个node生成一条路径
    {
        List<Integer> lists = new LinkedList<>(); //利用链表插入提高效率
        while(sn != null)
        {
            lists.add(0, sn.val);
            sn = sn.pre;
        }
        resultLists.add(lists);
    }
    return resultLists;
}

 public class SuperNode {
     int val;
     SuperNode pre;//指向前驱节点
     public SuperNode(int x)
        {
            val = x;
            pre = null;
        }
 }

//找到指定路径 返回null为没有 否则可以根据node遍历得到逆序
public void findPathSum(TreeNode root, int sum, SuperNode sn, List<SuperNode> superNodes) {
    if(root == null)
    {
        //空树  结束条件之一
        return;
    }
    else
    {
        if(root.left==null && root.right==null)
        {
            //叶子节点 left right都为空
            if(sum== root.val)
            {
                superNodes.add(sn);
                return;
            }
            else
            {
                return;
            }
        }
        else
        {
            //非叶子结点
            if(root.left != null)
            {
                SuperNode node = new SuperNode(root.left.val);
                node.pre = sn;
                findPathSum(root.left, sum-root.val,node, superNodes);

            }
            if(root.right != null)
            {
                SuperNode node = new SuperNode(root.right.val);
                node.pre = sn;
                findPathSum(root.right, sum-root.val,node, superNodes);

            }

        }
    }
}
时间: 2024-10-10 21:09:09

path_sum2的相关文章

path_sum

题目描述 Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum. For example: Given the below binary tree and sum = 22, 5 / 4 8 / / 11 13 4 / \ 7 2 1 return true