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

最新推荐文章于 2022-06-13 16:11:07 发布

MD_

最新推荐文章于 2022-06-13 16:11:07 发布

阅读量207

点赞数

分类专栏： leetcode 图搜索

原文链接：https://www.cnblogs.com/jundima/p/10176815.html

版权

leetcode 同时被 2 个专栏收录

189 篇文章 1 订阅

订阅专栏

图搜索

12 篇文章 0 订阅

订阅专栏

转载自

文章目录

问题描述
解题报告
实现代码

问题描述

解题报告

存储每个节点的父节点，然后以 $t a r g e t$ 为中心，从三个方向进行遍历【父节点、左节点、右节点】。

实现代码

//存储每个节点的父节点
class Solution{
 public:
     map<TreeNode*, TreeNode*>par;
     void findParent(TreeNode* root){
        if(!root) return;
        if(root->left){
            par[root->left]=root;
            findParent(root->left);
        }
        if(root->right){
            par[root->right]=root;
            findParent(root->right);
        }
     }
    //从三个方向进行遍历，注意边界条件
     void dfs(TreeNode* root, int k, set<TreeNode*> &visited,vector<int>& res){
         //在已访问过的元素中找到了root，这条路上的递归结束
         if(visited.find(root)!=visited.end())
            return;
         visited.insert(root);
         if(k==0){
            res.push_back(root->val);
            return;
         }
         if(root->left)
            dfs(root->left,k-1,visited,res);
         if(root->right)
            dfs(root->right,k-1,visited,res);
         TreeNode* p=par[root];
         if(p)
            dfs(p,k-1,visited,res);
     }
     vector<int> distanceK(TreeNode* root, TreeNode* target, int K){
        if(!root)
            return {};
        findParent(root);
        vector<int>res;
        set<TreeNode*>visited;
        dfs(target,K,visited,res);
        return res;
    }
};