LeetCode——863.二叉树中所有距离为k的结点

二叉树中所有距离为k的结点

---

题目：

给定一个二叉树（具有根结点 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” 实际上是树上的结点。
上面的输入仅仅是对这些对象进行了序列化描述。

提示：

给定的树是非空的，且最多有 K 个结点。
树上的每个结点都具有唯一的值 0 <= node.val <= 500 。
目标结点 target 是树上的结点。
0 <= K <= 1000.

---

思路:此题用到的算法为深度优先搜索，其主要考察了对递归函数的理解，该题难度较大且解决过程比较复杂。我使用了两个递归函数进行遍历，进而查找符合条件的结点。大致实现过程如下：

class Solution {

    List<Integer> res;
    TreeNode target;
    int K;

    public List<Integer> distanceK(TreeNode root, TreeNode target, int K) {
        res = new LinkedList<>();
        this.target = target;
        this.K = K;
        dfs(root);
        return res;
    }

    private int dfs(TreeNode root) {
        if (root == null) return -1;
        if (root == target) {
            subAdd(root, 0);
            return 1;
        } else {
            int L = dfs(root.left);// 计算左边的值
            int R = dfs(root.right);// 计算右边的值
            // 如果左边的不为-1，且为K，则此时计算右边与自己之间相差K-L-1的点
            if (L != -1) {
                if (L == K) res.add(root.val);
                subAdd(root.right, L + 1);
                return L + 1;
                // 否则判断左边不为-1，且为K，则此时计算左边与自己之间相差K-R-1的点
            } else if (R != -1) {
                if (R == K) res.add(root.val);
                subAdd(root.left, R + 1);
                return R + 1;
            }
        }
        return -1;
    }
// 使用递归的思想找到距离为k的结点，并储存到链表res中
    public void subAdd(TreeNode node, int dist) {
        if (node == null) return;
        if (dist == K) {
            res.add(node.val);
        } else {
            subAdd(node.left, dist + 1);
            subAdd(node.right, dist + 1);
        }
    }
}

附上GitHub源码