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

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

二叉树可以看做是无向图，因此根据二叉树建立邻接表，邻接表的结构是当前节点与其相邻的节点。构建邻接表时需要遍历二叉树的左右子树，由于建立的是无向图，需要在节点与及其根节点以及根节点与节点之间建立一条边。接下来需要使用深度优先遍历一层层往下遍历，遍历到一个点 k 值减少一，当然遍历到的点不能是节点本身，如果 k 减少到0，则将该值加入结果集。

/**
 * 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<TreeNode*,vector<TreeNode*>> map;
    vector<int> ans;
    void dfs(TreeNode* root)
    {
        if(root->left)
        {
            map[root].push_back(root->left);
            map[root->left].push_back(root);
            dfs(root->left);
        }
         if(root->right)
        {
            map[root].push_back(root->right);
            map[root->right].push_back(root);
            dfs(root->right);
        }
    }
    void dfs1(TreeNode* root,TreeNode* father,int k)
    {
        if(!k) ans.push_back(root->val);
        else 
        {
            for(auto son:map[root])
            {
                if(son!=father)
                {
                    dfs1(son,root,k-1);
                }
            }
        }
    }
    vector<int> distanceK(TreeNode* root, TreeNode* target, int k) {
        dfs(root);
        dfs1(target,NULL,k);
        return ans;
    }
};