力扣863
给定一个二叉树(具有根结点 root), 一个目标结点 target ,和一个整数值 K 。
返回到目标结点 target 距离为 K 的所有结点的值的列表。 答案可以以任何顺序返回。
示例 1:
输入:root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2
输出:[7,4,1]
解释:
所求结点为与目标结点(值为 5)距离为 2 的结点,
值分别为 7,4,以及 1
注意,输入的 "root" 和 "target" 实际上是树上的结点。
上面的输入仅仅是对这些对象进行了序列化描述。
 
提示:
给定的树是非空的,且最多有 K 个结点。
树上的每个结点都具有唯一的值 0 <= node.val <= 500 。
目标结点 target 是树上的结点。
0 <= K <= 1000.
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/all-nodes-distance-k-in-binary-tree
哈夫曼树!!!
此题需要找到和指定节点距离为固定值的全部节点,从后往前找好找,但是要找到父节点其它分支的满足题意得节点很困难。故本题采用从前向后的思想,记录从根节点出发,记录到达每一个节点所走的路径。向左走记为"1",向右走记为"3"。比较两个路径,在前几个相同路径中,两条路重合,二者距离为0。当出现分叉的情况,一个向左走,一个向右走,从这一刻开始,二者每一个条路每走一步都会使二者间距加一。所以,通过比较路径字符串,可以获得二者间距。此外,对于每一个节点的路径都等于他父节点路径加上左或右的标记,为了便于构造父子的联系,采用层序遍历。并将父节点的路径存入Map中。
1 package Leetcode; 2 3 import java.util.*; 4 5 public class distanceK { 6 public List<Integer> distanceK(TreeNode root, TreeNode target, int K) { 7 HashMap<TreeNode,String> map = new HashMap<>(); 8 Queue<TreeNode> queue = new LinkedList<>(); 9 List<Integer> res = new LinkedList<Integer>(); 10 queue.add(root); 11 map.put(root,"1"); 12 while (queue.size()!=0){ 13 int num = queue.size(); 14 for (int i = 0; i < num; i++) { 15 TreeNode father = queue.poll(); 16 if(father.left!=null){ 17 queue.add(father.left); 18 map.put(father.left,map.get(father)+1); 19 } 20 if(father.right!=null){ 21 queue.add(father.right); 22 map.put(father.right,map.get(father)+3); 23 } 24 } 25 } 26 String s = map.get(target); 27 for (Map.Entry<TreeNode,String> entry:map.entrySet()) { 28 if(check(s,entry.getValue())==K){ 29 res.add(entry.getKey().val); 30 } 31 } 32 return res; 33 } 34 private int check(String s, String cur) { 35 int minLen= Math.min(s.length(),cur.length()); 36 boolean flag = false; 37 int i = 0; 38 int way = 0; 39 for (; i <minLen ; i++) { 40 if(s.charAt(i)!=cur.charAt(i)){ 41 break; 42 } 43 } 44 if(i==minLen){way= Math.max(s.length(),cur.length())-Math.min(s.length(),cur.length());} 45 else{way = s.length()+cur.length()-2*i;} 46 return way; 47 } 48 49 public static void main(String[] args) { 50 TreeNode t1 = new TreeNode(3); 51 TreeNode t2 = new TreeNode(4); 52 TreeNode t3 = new TreeNode(5); 53 TreeNode t4 = new TreeNode(6); 54 TreeNode t5 = new TreeNode(7); 55 t1.left = t2; 56 t1.right = t3; 57 t2.left = t4; 58 t3.right = t5; 59 distanceK d = new distanceK(); 60 d.distanceK(t1,t4,3); 61 } 62 }
 
                    
                     
                    
                 
                    
                
 
                
            
         
         浙公网安备 33010602011771号
浙公网安备 33010602011771号