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

最新推荐文章于 2020-12-22 19:52:37 发布

_miccretti

最新推荐文章于 2020-12-22 19:52:37 发布

阅读量104

点赞数

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

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

版权

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;
        }
}