二叉树向父节点宽搜的方法

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

题目

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

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

示例

输入：
输出：
解释：所求结点为与目标结点（值为 5）距离为 2 的结点，值分别为 7，4，以及 1。

提示

给定的树是非空的。
树上的每个结点都具有唯一的值  。
目标结点 target 是树上的结点。

思路

求某个节点到其它节点的距离，存在向子节点搜索和向父节点搜索两种情况。向子节点搜索直接bfs就行了，但是向父节点搜索由于不能直接得到一个节点的父节点，所以需要预处理出一个哈希表存下每个节点对应的父节点，这里使用dfs即可。根据提示，由于每个节点的值是唯一的，所以可以使用每个节点的值作为键。

这题还有一个关键点是如何防止走回头路。使用一个节点记录每次bfs的源节点from，那么能够bfs的条件是该次bfs的下一个节点不是from。

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

    public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
        dfs(root,null);
        bfs(target,null,0,k);
        return res;
    }

    public void dfs(TreeNode root,TreeNode fa) {
        map.put(root.val,fa);
        if(root.left != null) {
            dfs(root.left,root);
        }
        if(root.right != null) {
            dfs(root.right,root);
        }
    }

    public void bfs(TreeNode root,TreeNode from,int depth,int k) {
        if(root == null) {
            return;
        }
        if(depth == k) {
            res.add(root.val);
            return;
        }

        if(root.left != from) {
            bfs(root.left,root,depth + 1,k);
        }
        if(root.right != from) {
            bfs(root.right,root,depth + 1,k);
        }
        if(map.get(root.val) != from) {
            bfs(map.get(root.val),root,depth + 1,k);
        }
    }
}