算法记录
LeetCode 题目:
给定一个二叉树(具有根结点 root), 一个目标结点 target ,和一个整数值 K 。返回到目标结点 target
距离为 K
的 所有结点的值的列表。 答案可以以任何顺序返回。
说明
一、题目
输入:root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2。输出:[7,4,1]
二、分析
- 求解的是距离目标点的一定距离的节点的集合,也就是说不仅需要遍历当前的子节点,还需要判断的父节点及兄弟节点的距离关系。
- 首先需要遍历拿到每个节点的父节点信息,因为二叉树只保留了孩子节点的信息,父节点需要自己去遍历建立。
- 利用递归的思想首先遍历左节点、然后右节点、最后父节点,为了防止节点来回遍历,需要指定不等的条件 下一个需要开始的节点不能是上一次已经出发的节点。
- 当达到指定的距离步数时则可将节点信息并入到列表当中,最后结束返回结果列表。
class Solution {
TreeNode[] parents = new TreeNode[501];
List<Integer> result = new ArrayList<Integer>();
public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
findParents(root);
findAns(target, null, k);
return result;
}
public void findParents(TreeNode node) {
if (node.left != null) {
parents[node.left.val] = node;
findParents(node.left);
}
if (node.right != null) {
parents[node.right.val] = node;
findParents(node.right);
}
}
public void findAns(TreeNode node, TreeNode from, int k) {
if (node == null) {
return;
}
if (0 == k) {
result.add(node.val);
return;
}
if (node.left != from) {
findAns(node.left, node, k-1);
}
if (node.right != from) {
findAns(node.right, node, k-1);
}
if (parents[node.val] != from) {
findAns(parents[node.val], node, k-1);
}
}
}
总结
熟悉二叉树之间的连接关系和遍历方法。