LeetCode——993 二叉树的堂兄弟节点(JAVA)

最新推荐文章于 2023-04-06 08:00:00 发布

囷囷

最新推荐文章于 2023-04-06 08:00:00 发布

阅读量225

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：二叉树 java 算法 leetcode 数据结构

本文链接：https://blog.csdn.net/weixin_42257812/article/details/116357013

LeetCode 专栏收录该内容

52 篇文章

订阅专栏

本文介绍了一种方法来确定二叉树中两个特定节点是否为堂兄弟节点。通过广度优先搜索（BFS）遍历树结构，并记录每个节点的深度和父节点，最终比较目标节点的深度及父节点来判断是否符合堂兄弟条件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在二叉树中，根节点位于深度 0处，每个深度为 k的节点的子节点位于深度 k+1处。

如果二叉树的两个节点深度相同，但 父节点不同 ，则它们是一对堂兄弟节点。

我们给出了具有唯一值的二叉树的根节点 root，以及树中两个不同节点的值 x和 y。

只有与值 x和 y对应的节点是堂兄弟节点时，才返回 true。否则，返回 false。

示例 1：

在这里插入图片描述

输入：root = [1,2,3,4], x = 4, y = 3
输出：false

示例 2：

在这里插入图片描述

输入：root = [1,2,3,null,4,null,5], x = 5, y = 4
输出：true

示例 3：

在这里插入图片描述

输入：root = [1,2,3,null,4], x = 2, y = 3
输出：false

提示：

二叉树的节点数介于 2 到 100 之间。
每个节点的值都是唯一的、范围为 1 到 100 的整数。

思路

本题的思路主要是从堂兄弟这个定义上入手：1.查询的两个结点深度相同；2.它们的父结点不同。

因此，需要新定义一个类，将结点的深度和它的父结点情况这两个信息和它本身绑定在一起（我这里写了Node类）。然后，做一次BFS搜索，搜索的时候记录结点本身的深度情况和它的父结点（根结点的父结点为null）。因为BFS会搜索所有结点，所以当BFS结束时，所有结点的信息都已建立好，只需要根据所要查询的结点判断即可。

因为输入的查询值是结点的val信息，并不能直接获得结点本身。因此，我建立了一个HashMap，来建立val->Node这样一个映射，所以，在输入val信息之后，通过HashMap即可获得对应的结点。

【注】在判断父结点的时候，要先判断父结点存在（如果父结点为null，进行xFather!=yFather判断时会报错，另外，若父结点为null，说明该结点为根结点，直接返回false即可）。

代码

public class Solution {
	public class TreeNode {
		int val;
		TreeNode left;
		TreeNode right;
		TreeNode() {}
		TreeNode(int val) { this.val = val; }
		TreeNode(int val, TreeNode left, TreeNode right) {
			this.val = val;
			this.left = left;
			this.right = right;
		}
	}
	class Node{
		TreeNode node;
		int layer;//记录结点所在层次
		Node father;//记录父结点
		Node(TreeNode node, int layer, Node father){
			this.node = node;
			this.layer = layer;
			this.father = father;
		}
	}
	Queue<Node> queue = new LinkedList<Node>();
	HashMap<Integer, Node> hashMap = new HashMap<Integer, Node>();
	public void bfs(Node root) {
		queue.offer(root);
		while(queue.isEmpty()==false) {
			Node topNode = queue.poll();//出队,并返回队首元素
			hashMap.put(topNode.node.val, topNode);//建立val->Node的映射
			if(topNode.node.left!=null) {
				Node lchild = new Node(topNode.node.left, topNode.layer+1, topNode);
				queue.offer(lchild);
			}
			if(topNode.node.right!=null) {
				Node rchild = new Node(topNode.node.right, topNode.layer+1, topNode);
				queue.offer(rchild);
			}
		}
	}
	public boolean isCousins(TreeNode root, int x, int y) {
		Node rootNode = new Node(root, 0, null);
		bfs(rootNode);
		Node xx = hashMap.get(x);
		Node yy = hashMap.get(y);
		Node xFather = xx.father;
		Node yFather = yy.father;
		if(xFather!=null&&yFather!=null) {
			if(xFather!=yFather&&xx.layer==yy.layer) return true;
			else return false;
		}
		else return false;
	}
}