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

题目

给定一个二叉树（具有根结点 root）， 一个目标结点 target ，和一个整数值 K 。返回到目标结点 target 距离为 K 的所有结点的值的列表。 答案可以以任何顺序返回。
样例：

提示：

还是老样子， 先想这道题利用以下距离目标值的距离是k的节点， 是他的字节点好说， 直接遍历就好了， 但是它的父节点怎么统计。

• 思路： 首先它的父节点肯定还有右子节点或者还有父节点， 如何从当前节点得到父节点， 那么就用parents先记录以下每个节点的父节点， 遍历的时候从target开始遍历， 遍历左右节点， 以及父节点， 带着距离， 最后判断距离是否够k就可以了，
• 二： 建图利用bfs就可以了或者dfs都可以那个方法较为简单就不写了。

/**
 * 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*, TreeNode*> parents;
    vector<int> res;
    void getParents(TreeNode * root){

       if (root -> left != NULL){
           parents[root -> left] = root;
           getParents (root -> left);
       }
       if (root -> right){
           parents[root -> right] = root;
           getParents (root -> right);
       }
    }

    void findD (TreeNode* t, TreeNode *from, int dep, int k){
        
        if (t == NULL) return;
        if (dep == k){
            res.push_back (t -> val);
        }

        if (t -> left != from)
            findD (t -> left, t, dep + 1, k);
        if (t -> right != from)
            findD (t -> right, t, dep + 1, k);
        if (parents[t] != from)
            findD (parents[t], t, dep + 1, k);        

    }


    vector<int> distanceK(TreeNode* root, TreeNode* target, int k) {
        getParents(root);
        findD (target, NULL, 0, k);
        return res;
    }
};