一、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
- 从target结点开始dfs(target, k),找出所有target子结点中所有满足题意的结点,加入到list中。
- 找出target的父结点,执行dfs(target, k - 1),找出所有没有访问过的满足题意的结点。
- 找出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);
}
}
}