LeetCode＜每日一题＞二叉树中所有距离为 K 的结点

最新推荐文章于 2022-10-06 18:37:40 发布

酱酱熊

最新推荐文章于 2022-10-06 18:37:40 发布

阅读量87

点赞数

分类专栏：算法文章标签：算法数据结构

本文链接：https://blog.csdn.net/weixin_42496727/article/details/119173190

版权

算法专栏收录该内容

136 篇文章 1 订阅

订阅专栏

该博客介绍了一种利用深度优先搜索（DFS）解决二叉树问题的方法，特别是寻找目标节点周围特定距离的节点。通过建立节点到父节点的映射，并使用标志数组避免重复访问，算法能够有效地遍历树并在给定距离内找到目标节点。

摘要由CSDN通过智能技术生成

题目链接
思路：变相的深搜，也就是将父节点也加进去进行深度遍历

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {

    List<Integer> distanceK_res = new ArrayList<>();//记录答案
    HashMap<Integer, TreeNode> distanceK_map = new HashMap<>();//记录父节点  node -> parent
    boolean[] flags = new boolean[501];// 记录是否被用过了
    public List<Integer> distanceK(TreeNode root, TreeNode target, int k) {
        if(k <= 500){
            distanceKHelper1(root);
            distanceKHelper2(target, k);
        }
        return distanceK_res;
    }

    public void distanceKHelper2(TreeNode target, int k){
        flags[target.val] = true;
        if(k==0){
            distanceK_res.add(target.val);
            return;
        }
        if( target.left != null && !flags[target.left.val]){
            flags[target.left.val] = true;
            distanceKHelper2(target.left, k-1);
        }
        if(target.right != null && !flags[target.right.val]){
            flags[target.right.val] = true;
            distanceKHelper2(target.right, k-1);
        }

        TreeNode parent = distanceK_map.get(target.val);
        if( parent!= null && !flags[parent.val]){
            distanceKHelper2(parent, k-1);
        }

    }

    public void distanceKHelper1(TreeNode root){
        if(root.left != null){
            distanceK_map.put(root.left.val, root);
            distanceKHelper1(root.left);
        }
        if(root.right != null){
            distanceK_map.put(root.right.val, root);
            distanceKHelper1(root.right);
        }
    }
}