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

最新推荐文章于 2023-05-14 15:05:47 发布

Gogo-2020

最新推荐文章于 2023-05-14 15:05:47 发布

阅读量105

点赞数

分类专栏： leetcode 文章标签：二叉树算法 java 数据结构

本文链接：https://blog.csdn.net/weixin_44560620/article/details/119184267

版权

给定一个二叉树、目标节点和整数K，该博客介绍了如何找到与目标节点距离为K的所有节点。通过将二叉树转化为无向图，并进行广度优先搜索来解决这个问题。示例展示了当目标节点值为5，K为2时，返回值为[7,4,1]的解题过程。" 111722036,10293909,Python数值转换与特征工程实战,"['Python', '数据预处理', '特征工程']

摘要由CSDN通过智能技术生成

给定一个二叉树（具有根结点 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.

解题思路

因为二叉树就是一种特殊的有向图，因此我们可以遍历树，重构一个无向图
通过对无向图进行广度优先搜索，以target节点为起点，找出距离为k的所有点

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
   
    public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
   

        Map<TreeNode,List<TreeNode>> map=new HashMap<>();
        List<Integer> list=new ArrayList<><