力扣（7/28每日一题）二叉树中所有距离为 K 的结点

最新推荐文章于 2024-10-08 12:54:42 发布

Jiawen9

最新推荐文章于 2024-10-08 12:54:42 发布

阅读量151

点赞数

文章标签： c++ leetcode dfs

本文链接：https://blog.csdn.net/cjw838982809/article/details/119181769

版权

该博客介绍了如何解决一个LeetCode问题，即在给定的二叉树中找到与目标节点距离为K的所有节点的值。算法使用深度优先搜索（DFS）和哈希表记录节点的父节点，以有效地找到所需距离的节点。示例展示了如何在给定的二叉树结构中找到距离目标节点两步的所有节点值。

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

给定一个二叉树（具有根结点 root），一个目标结点 target ，和一个整数值 K 。

返回到目标结点 target 距离为 K 的所有结点的值的列表。答案可以以任何顺序返回。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/all-nodes-distance-k-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例 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" 实际上是树上的结点。
上面的输入仅仅是对这些对象进行了序列化描述。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/all-nodes-distance-k-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

#include<iostream>
using namespace std

unordered_map<int,TreeNode*> parents;
vector<int> ans;

//用一个哈希表记录所有结点的父结点
void findParents(TreeNode* root){
    if(root->left != nullptr){
        parents[root->left->val] = root;
        findParents(root->left);
    }
    if(root->right != nullptr){
        parents[root->right->val] = root;
        findParents(root->right);
    }
}

//深搜
void findAns(TreeNode* root, TreeNode* from, int depth, int k){
    if(root == nullptr) return;
    if(depth == k){
        ans.push_back(root->val);
        return;
    }
    if(root->left != from) findAns(root->left, root, depth + 1, k);
    if(root->right != from) findAns(root->right, root, depth + 1, k);
    if(parents[root->val] != from) findAns(parents[root->val], root, depth + 1, k);
}

vector<int> distanceK(TreeNode* root, TreeNode* target, int k) {
    //dfs记录所有父结点
    findParents(root);
    findAns(target, nullptr, 0, k);
    return ans;
}

运行结果：