给定一个二叉树(具有根结点 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
示例 2:
输入: root = [1], target = 1, k = 3
输出: []
提示:
节点数在 [1, 500] 范围内
0 <= Node.val <= 500
Node.val 中所有值 不同
目标结点 target 是树上的结点。
0 <= k <= 1000
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/all-nodes-distance-k-in-binary-tree
思路:
深度遍历 1. 目标结点 target 的父节点也有可能有距离为 k 的节点,所以我们需要记录一下每个节点的父节点,以便于能追溯父节点处的距离为 k 的节点 a. 定义一个 map,key 为当前节点的值,value 为父节点 2. 定义 findParants 函数,深度遍历二叉树查找所有节点的父亲节点 3. 定义 findDistancek 函数,查找所有距离 目标结点 target 为 k 的所有节点 4. 为了防止走回头路重复计算节点,需要判断当前节点下一个遍历的节点是否是不是遍历当前节点的上一个节点
java代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
Map<Integer,TreeNode> parantsMap = new HashMap<>();
findParants(root,null,parantsMap);
Set<Integer> resultSet = new HashSet<>();
findDistancek(target,target,k,0,parantsMap,resultSet);
return resultSet.stream().toList();
}
public void findDistancek(TreeNode currentNode,TreeNode from,int k,int depth,Map<Integer,TreeNode> parantsMap,Set<Integer> result){
if(currentNode==null) {
return;
}
if(depth==k) {
result.add(currentNode.val);
return;
}
if(currentNode.left!=null && currentNode.left.val != from.val) {
findDistancek(currentNode.left,currentNode,k,depth+1,parantsMap,result);
}
if(currentNode.right!=null && currentNode.right.val != from.val) {
findDistancek(currentNode.right,currentNode,k,depth+1,parantsMap,result);
}
if(parantsMap.get(currentNode.val) != null && parantsMap.get(currentNode.val).val != from.val) {
findDistancek(parantsMap.get(currentNode.val),currentNode,k,depth+1,parantsMap,result);
}
}
public void findParants(TreeNode root,TreeNode parant,Map<Integer,TreeNode> map){
if(root==null) {
return;
}
if(parant != null) {
map.put(root.val,parant);
}
findParants(root.left,root,map);
findParants(root.right,root,map);
}
}