LeetCode——113. 路径总和 II

最新推荐文章于 2022-08-13 10:15:45 发布

豌豆二锅头

最新推荐文章于 2022-08-13 10:15:45 发布

阅读量59

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/weixin_44750790/article/details/119673590

版权

LeetCode 专栏收录该内容

160 篇文章 1 订阅

订阅专栏

该博客讨论了一种解决二叉树路径和问题的算法。通过创建一个新类型存储节点及其路径和，并使用哈希表保存节点与其父节点的关系，博主详细解释了如何遍历树来找到所有路径总和等于目标和的叶子节点路径。在遍历过程中，当找到目标路径时，回溯父节点以获取完整路径并将其添加到结果列表中。注意在遍历过程中要保存当前节点的副本，以避免节点值在后续逻辑中被修改。

摘要由CSDN通过智能技术生成

题目

给你二叉树的根节点 root 和一个整数目标和 targetSum ，找出所有从根节点到叶子节点路径总和等于给定目标和的路径。

叶子节点是指没有子节点的节点。
题目传送门

思路

新开一个类型，存放当前节点和当前节点的路径和，并且新开一个hashmap，把当前的节点和对应的父节点以哈希表的形式存放到哈希表中，一旦遍历到了当前节点的路径和等于目标的路径和，那么就不断去获取它的父节点直到根节点，并且将路径加入到列表中即可！

代码

class Element
 {
     TreeNode node;
     int sum;
     public Element(TreeNode node,int sum)
     {
         this.node=node;
         this.sum=sum;
     }
 }
class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int targetSum)
    {
        List<List<Integer>> ans =new LinkedList<>();
        if(root==null)
            return ans;
        // 子节点在前，父节点在后    
        HashMap<TreeNode,TreeNode>map = new HashMap<>();
            map.put(root,null);
        // 搜索队列    
        Queue<Element> queue = new LinkedList<>();
            queue.add(new Element(root,root.val));
        
        while(!queue.isEmpty())
        {
            Element element = queue.poll();
            TreeNode temp = element.node;
            TreeNode node =element.node;   // 多获取一次当前节点，防止因为进入了目标路径而当前节点被改变造成下面的代码不符合逻辑的情况发生
            if(temp.left==null && temp.right==null && element.sum==targetSum)
            {
                List<Integer> list = new LinkedList<>();
                list.add(0,temp.val);
                while(map.get(temp)!=null)
                {
                    temp=map.get(temp);
                    list.add(0,temp.val);
                }
                ans.add(list);
            }
            if(node.left!=null)
            {
                map.put(node.left,node);  // 存放父节点
                queue.add(new Element(node.left,element.sum+node.left.val));
            }
            if(node.right!=null)
            {
                map.put(node.right,node); // 存放父节点
                queue.add(new Element(node.right,element.sum+node.right.val));
            }
        }
        return ans;
    }
}