863. 二叉树中所有距离为 K 的结点

Posted 2021-08-29 易小顺

算法记录

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

---

总结

熟悉二叉树之间的连接关系和遍历方法。