最近刷LeetCode题目的一些思路,题目信息
给定一个二叉树(具有根结点 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" 实际上是树上的结点。 上面的输入仅仅是对这些对象进行了序列化描述。
提示:
- 给定的树是非空的,且最多有
K
个结点。 - 树上的每个结点都具有唯一的值
0 <= node.val <= 500
。 - 目标结点
target
是树上的结点。 0 <= K <= 1000
.
--------------------------------------------------------------------------------------
因为目标结点可能位于任何位置,而二叉树又是单向的,所以需要先将所有树结点串联起来,指明每一个结点的父结点,然后根据目标结点向其子结点父节点扩散查找。
1. 整理数据,给每个结点指定父结点,然后保存在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); }
2. 以目标结点为中心,进行扩散遍历
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; }