【前缀思想】二叉树中所有距离为 K 的结点

最新推荐文章于 2020-05-11 13:36:43 发布

_miccretti

最新推荐文章于 2020-05-11 13:36:43 发布

阅读量132

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/yuelien/p/10532747.html

本文介绍了一种使用哈希映射实现的简单方法，来解决二叉树中寻找所有与目标节点距离为K的节点的问题。通过递归地构建每个节点到根的路径编码，可以高效地找出所有符合条件的节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

863. 二叉树中所有距离为 K 的结点

class Solution {
     Map<TreeNode,String>map=new HashMap<>();
     String path;
        void getNodeDist(TreeNode root,TreeNode target,String p){
            if(root!=null){
                path=root==target?p:path;
                map.put(root, p);
                getNodeDist(root.left,target,p+"0");
                getNodeDist(root.right,target,p+"1");
            }
        }
        public List<Integer> distanceK(TreeNode root, TreeNode target, int K) {
            List<Integer>list=new ArrayList<>();
            getNodeDist(root,target,"");int i;
            for(TreeNode key:map.keySet()){
                String s=map.get(key);
                for(i=0;i<s.length()&&i<path.length()&&s.charAt(i)==path.charAt(i);i++);
                    if(s.length()-i+path.length()-i==K)
                        list.add(key.val);
            }
            return list;
        }
}