二叉树中所有距离为K的节点（DFS+BFS）

谁借莪１个温暖的怀抱￠ 2023-10-06 14:02 19阅读 0赞

## 二叉树中所有距离为K的节点（DFS+BFS） ##

#### 问题： ####

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 1]

#### 思路： ####

为找到与节点距离为K的节点，需解决节点回溯到其父节点的问题，因此需知道每个节点的父节点，使用哈希表进行保存，通过DFS建立此哈希表

将target节点push到队列中，进行三个方向的BFS（左孩子、右孩子、父节点），为防止重复访问节点，需使用set保存已访问过的节点

#### 代码： ####

/**
     * 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 {
        
    private:
        unordered_map< TreeNode*,TreeNode* > table;
    public:
        void dfs( TreeNode* root,TreeNode* pre )
        {
        
            if( root )
            {
        
                table[root]=pre;
                dfs( root->left,root );
                dfs( root->right,root );
            }
        }
        vector<int> distanceK(TreeNode* root, TreeNode* target, int K) {
        
            dfs( root,NULL );
            queue< TreeNode* > q;
            TreeNode* p=target;
            q.push( p );
            vector<int> ans;
            unordered_set< TreeNode* > visited;
            int k=0;
            while( !q.empty() )
            {
           
                int size=q.size();
                while( size-- )
                {
           
                    p=q.front();
                    if( k==K )
                        ans.push_back( p->val );
                    q.pop();
                    visited.insert( p );
                    if( p->left && visited.count( p->left )==0 )
                    {
        
                        q.push( p->left );
                        visited.insert( p->left );
                    }  
                    if( p->right && visited.count( p->right )==0 )
                    {
        
                         q.push( p->right );
                         visited.insert( p->right );
                    }
                    if( table.count( p )==1 && visited.count( table[p] )==0 && table[p]!=NULL )
                    {
        
                        q.push( table[p] );
                        visited.insert( table[p] );
                    }          
                }
                k++;
            }
            return ans;
        }
    };

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200801091033607.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200801091043860.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70