本题来自力扣第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);
}
}
}
总结
对于一个看起来好像只涉及树的问题,当其复杂到一定程度,其实可以转换为图的问题来解决,往往以为有难度的问题就会因此迎刃而解。