力扣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 }

 

posted @ 2020-02-28 20:33  HEUzbc  阅读(300)  评论(0)    收藏  举报