LeetCode863—二叉树中所有距离为K的结点（java版）

题目描述：

标签：树  深度优先搜索  广度优先搜索  二叉树

给定一个二叉树（具有根结点 root）， 一个目标结点 target ，和一个整数值 K 。

返回到目标结点 target 距离为 K 的所有结点的值的列表。 答案可以以任何顺序返回。

提示：

1、给定的树是非空的。
2、树上的每个结点都具有唯一的值 0 <= node.val <= 500 。
3、目标结点 target 是树上的结点。
4、0 <= K <= 1000.

 代码：

思路分析：

以target为起点，去找距离为k的结点。因为距离为k的结点可能要回溯到上层结点，所以得先用map保存各个结点的父节点（由于每个结点具有唯一的值，所以可以用结点的值作为key）。

1、定义一个函数getParents()用来将各个结点的父结点放入map中

        判断左节点是否为null，不是的话就map.put(node.left.val, node)，继续递归getParents(node.left)

        判断右节点是否为null，不是的话就map.put(node.left.val, node)，继续递归getParents(node.right)

2、定义一个函数getAns()，找到距离为k的结点添加到list中，其中为了防止重复判断同一个节点，所以定义了一个源节点from

        1）、递归结束条件，node==null 直接返回或者 depth==k 添加结点后返回

        2）、判断左节点是不是源节点，不是的话继续递归getAns()

        3）、判断右节点是不是源节点，不是的话继续递归getAns()

        4）、判断当前结点的父结点是不是源节点，不是的话往父结点继续递归getAns()

/**
 * 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> parents = new HashMap<>();
    List<Integer> result = new ArrayList<>();

    public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
        getParents(root);
        getAns(target, null, 0, k);
        return result;
    }

    public void getParents(TreeNode node){
        if(node.left != null){
            parents.put(node.left.val,node);
            getParents(node.left);
        }
        if(node.right != null){
            parents.put(node.right.val,node);
            getParents(node.right);
        }
    }

    public void getAns(TreeNode node, TreeNode from, int depth, int k){
        if(node == null){
            return;
        }
        if(depth == k){
            result.add(node.val);
            return;
        }
        if(node.left != from){
            getAns(node.left, node, depth+1, k);
        }
        if(node.right != from){
            getAns(node.right, node, depth+1, k);
        }
        if(parents.get(node.val) != from){
            getAns(parents.get(node.val), node, depth+1, k);
        }
    }
}