leetcode863. 二叉树中所有距离为 K 的结点( 建图+BFS)

最新推荐文章于 2022-04-21 13:53:10 发布

Kerin637

最新推荐文章于 2022-04-21 13:53:10 发布

阅读量115

点赞数 2

分类专栏： Leetcode刷题文章标签：二叉树邻接表 BFS 图遍历层次距离

本文链接：https://blog.csdn.net/mds637/article/details/119175091

版权

Leetcode刷题专栏收录该内容

15 篇文章 0 订阅

订阅专栏

题目链接
在这里插入图片描述

根据题意很容易想到用从target结点开始进行dfs或者bfs但是因为给的是二叉树，父结点信息无法直接得到，所以要想办法把二叉树转化成邻接表来存储，事实上对于树的每一个结点，它的邻接点只是缺少一个父结点信息而已，鉴于树上的每个结点都具有唯一的值，因此可以建立一个map，key就是结点的val，值就是该结点的父结点，这样每个结点的邻接点都可以访问到了，对图进行dfs或者bfs就可以了。
这里我采用的是BFS，因为需要比较层次信息，于是设计了一个结构体存储结点和相应的层次，但是比较麻烦，看了大佬的方法，感觉更为简单，首先n=Q.size()记录队列大小（也就是同一层的结点个数），用for循环pop n次，循环结束再level++。
在这里插入图片描述
自己写的又臭又长的代码qwq

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    unordered_map<int,TreeNode *> parents;
    struct Node{
        TreeNode *node;
        int level;
    };
    bool vis[505]={false};
    void findParents(TreeNode *root){
        if(root->left!=NULL){
            parents[root->left->val] = root;
            findParents(root->left);
        }
        if(root->right!=NULL){
            parents[root->right->val] =root;
            findParents(root->right);
        }
    }
    vector<int> distanceK(TreeNode* root, TreeNode* target, int k) {
        //以target为基础进行BFS
        findParents(root);
        queue<struct Node> Q;
        struct Node tar;
        tar.node=target;
        tar.level=0;
        Q.push(tar);
        vis[(tar.node)->val]=true;
        vector<int> res;
        while(!Q.empty()){
            struct Node tmp=Q.front();
            Q.pop();
            vis[(tmp.node)->val]=true;
            if(tmp.level==k){
                res.push_back(tmp.node->val);
            }
            if((tmp.node)->left!=NULL && !vis[(tmp.node)->left->val]){
                struct Node left;
                left.node=(tmp.node)->left;
                left.level=tmp.level+1;
                Q.push(left);
            }// Q.push(struct Node((tmp.node)->left,tmp.level+1);
            if((tmp.node)->right!=NULL && !vis[(tmp.node)->right->val]) //Q.push(struct Node((tmp.node)->right,tmp.level+1);
            {
                struct Node right;
                right.node=(tmp.node)->right;
                right.level=tmp.level+1;
                Q.push(right);
            }
            if(tmp.node->val != root->val && parents[tmp.node->val]!=NULL && !vis[parents[tmp.node->val]->val]){
                struct Node p;
                p.node=parents[tmp.node->val];
                p.level=tmp.level+1;
                Q.push(p);
            }
        }
        return res;
    }
};