算法：二叉树中所有距离为 K 的结点 【树转图+BFS】

本题来自力扣第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” 实际上是树上的结点。
上面的输入仅仅是对这些对象进行了序列化描述。

提示

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

解题思路

由于对于树中的一个结点的距离问题，会存在跨子树等复杂的情况，如果单纯使用树的DFS或者BFS是不好实现的。而对于图而言，则是一个非常直观的问题，即求图中的某个点到其他任意一个点的距离符合要求的问题，采用图的BFS就能搞定。因此本题的解法可以先将树转为图，具体办法是采用哈希表的存储结构来实现。

代码

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

/** 树转图+BFS */
class Solution {

    private HashMap<TreeNode, HashSet<TreeNode>> map = new HashMap<>();

    public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
        treeToGraph(root, null);
        List<Integer> res = new ArrayList<>();
        if (k == 0) {
            res.add(target.val);
            return res;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(target);
        while (!queue.isEmpty()) {
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                HashSet<TreeNode> set = map.get(node);
                //将图中相邻结点到该结点的映射剔除，避免出现结点循环入队
                for (TreeNode temp : set) {
                    map.get(temp).remove(node);
                }
                queue.addAll(set);
            }
            k--;
            if (k <= 0) {
                while (!queue.isEmpty()) {
                    res.add(queue.poll().val);
                }
            }
        }
        return res;
    }

    private void treeToGraph(TreeNode root, TreeNode pre) {
        map.put(root, new HashSet<TreeNode>());
        //一个树结点的图结构：映射其父节点、左孩子结点、右孩子结点
        if (pre != null) map.get(root).add(pre);
        if (root.left != null) {
            map.get(root).add(root.left);
            treeToGraph(root.left, root);
        }
        if (root.right != null) {
            map.get(root).add(root.right);
            treeToGraph(root.right, root);
        }
    }
}

总结

对于一个看起来好像只涉及树的问题，当其复杂到一定程度，其实可以转换为图的问题来解决，往往以为有难度的问题就会因此迎刃而解。