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

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

中等难度

给定一个二叉树（具有根结点 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

示例 2:

输入: root = [1], target = 1, k = 3
输出: []

提示:

• 节点数在 [1, 500] 范围内
• 0 <= Node.val <= 500
• Node.val 中所有值 不同
• 目标结点 target 是树上的结点。
• 0 <= k <= 1000

思路

1、建哈希表 key=Integer ,value=TreeNode,如果某个节点的左孩子不为空，则将左孩子的值和父节点映射，存进哈希表。若右孩子不为空，则将右孩子的值和父节点映射，存进哈希表。建此表的目的是为了可以搜集每个节点的父节点，由示例1所示，我们需要向上递归，当查到target，可以根据target的值，对哈希表查询，查到父节点，直接跳转，这样就实现了向上递归

2、为了防止走回头路，所以设计了一个from标志，在向上递归的过程中，在当前访问节点的基础上去追它的父节点，而父节点在下一次的递归过程中还是会两个子节点都去访问的，而其中一个上一次已经访问过了，所以from存的是上次访问的子节点，这次无需再访问了，再访问必然会重复计算的

3、时间复杂度O(N)，空间复杂度O(N)

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {

    Map<Integer,TreeNode> parent = new HashMap<Integer,TreeNode>();
    List<Integer> res = new ArrayList<>();

    public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
        
        findParent(root);

        findAns(target,null,0,k);

        return res;
    }

    private void findParent(TreeNode node){
        if(node.left != null){
            parent.put(node.left.val,node);
            findParent(node.left);
        }

        if(node.right != null){
            parent.put(node.right.val ,node);
            findParent(node.right);
        }
    }

    private void findAns(TreeNode node, TreeNode from, int depth, int k){
        if(node == null){
            return;
        }

        if(depth == k){
            res.add(node.val);
        }

        if(node.left != from){
            findAns(node.left, node, depth+1 , k);
        }

        if(node.right != from){
            findAns(node.right,node,depth+1 , k);
        }

        if(parent.get(node.val) != from){
            findAns(parent.get(node.val), node, depth+1, k);
        }
    }
}