LeetCode-All Nodes Distance K in Binary Tree

一、Description

题目描述：给定一个二叉树，一个target结点，一个K，找出所有离target结点的距离为K的所有结点，返回一个list。

Example 1:

Input: root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2

Output: [7,4,1]

---

二、Analyzation

1. 从target结点开始dfs(target, k)，找出所有target子结点中所有满足题意的结点，加入到list中。
2. 找出target的父结点，执行dfs(target, k - 1)，找出所有没有访问过的满足题意的结点。
3. 找出target的父结点的父结点和父结点的其它子结点的父结点，执行dfs(target, k - 2)...以此内推，当k = 0时将当前访问结点加入到list中，递归回溯。

---

三、Accepted code

class Solution {
    Map<TreeNode, TreeNode> map = new HashMap<>();
    Set<TreeNode> set = new HashSet<>();
    List<Integer> list = new ArrayList<>();
    public List<Integer> distanceK(TreeNode root, TreeNode target, int K) {
        if (root == null) {
            return list;
        }
        findParent(root);
        dfs(target, K);
        return list;
    }
    public void findParent(TreeNode root) {
        if (root.left != null) {
            map.put(root.left, root);
            findParent(root.left);
        }
        if (root.right != null) {
            map.put(root.right, root);
            findParent(root.right);
        }
    }
    public void dfs(TreeNode node, int k) {
        if (set.contains(node)) {
            return ;
        }
        set.add(node);
        if (k == 0) {
            list.add(node.val);
        }
        if (node.left != null) {
            dfs(node.left, k - 1);
        }
        if (node.right != null) {
            dfs(node.right, k - 1);
        }
        TreeNode parent = map.get(node);
        if (parent != null) {
            dfs(parent, k - 1);
        }
    }
}