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);
}
}
}