Arithmetic5:二叉树中所有距离为K 的结点

最新推荐文章于 2020-11-23 10:18:10 发布

Ataraxiaone

最新推荐文章于 2020-11-23 10:18:10 发布

阅读量171

点赞数

分类专栏： Android 实验室文章标签： 2019.9.2

本文链接：https://blog.csdn.net/qq_41963701/article/details/100375572

版权

Android 实验室专栏收录该内容

6 篇文章 0 订阅

订阅专栏

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

算法概述：

描述：

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

示例：

在这里插入图片描述

输入：root = [3,5,1,6,2,0,8,null,null,7,4], target = 5, K = 2//所求结点为与目标结点（值为 5）距离为 2 的结点
输出：[7,4,1]

说明：

?给定的树是非空的，且最多有 K 个结点。
?树上的每个结点都具有唯一的值 0 <= node.val <= 500 。
?目标结点 target 是树上的结点。
?0 <= K <= 1000.

我的理解：

目标节点Target可能位于二叉树的任意位置，而二叉树又是单向的，所以需要先将所有的树节点串联起来，指明每个节点的父节点并保存在Map中，再根据目标节点向其子节点、父节点扩散遍历。

实现代码：

private void setParentTreeNode(TreeNode root, Map<TreeNode,TreeNode> parentTreeNodeMap, TreeNode parent){
    if(root == null){
        return;
    }
    if(parent != null){
        parentTreeNodeMap.put(root, parent);
    }
    setParentTreeNode(root.left,parentTreeNodeMap,root);
    setParentTreeNode(root.right,parentTreeNodeMap,root);
    }
public List<Integer> distanceK(TreeNode root, TreeNode target, int K) {
    List<Integer> retList=new ArrayList<>();

    Map<TreeNode,TreeNode> parentsTreeMap=new HashMap<>();
    setParentTreeNode(root, parentsTreeMap,null);

    Queue<TreeNode> checkQueue=new LinkedList<>();
    checkQueue.add(target);

    Set<TreeNode> hasCheck=new HashSet<>();
    hasCheck.add(target);

    while(!checkQueue.isEmpty()&&K>0){
        K--;
        int size=checkQueue.size();
        for(int i=0; i<size; i++){
            TreeNode treeNode=checkQueue.poll();
            if (treeNode.left != null && hasCheck.add(treeNode.left))
                checkQueue.add(treeNode.left);
            if (treeNode.right != null && hasCheck.add(treeNode.right))
                checkQueue.add(treeNode.right);
            if (parentsTreeMap.containsKey(treeNode) && hasCheck.add(parentsTreeMap.get(treeNode)))
                checkQueue.add(parentsTreeMap.get(treeNode));
        }
    }
    while (!checkQueue.isEmpty()){
        retList.add(checkQueue.poll().val);
    }
    return retList;
}