993. 二叉树的堂兄弟节点
在二叉树中,根节点位于深度 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 的整数。
题解:
我们先在二叉树中使用深搜搜索x,若搜索到了则存储搜到的x所在的深度,且存储其对应的父节点val,没搜索到不进行操作;
接着同理将y放进深搜中搜索其深度及其父节点对应的val.
然后比较二者是否为深度相同,val不同即可。
代码(C):
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int res[2] = {0};//不使用数组的话在存储时比较麻烦,使用数组直接按照下标存储即可
int val[2] = {0};
void dfs(struct TreeNode*root,int n,int h,int i,struct TreeNode*root1)
{
if(root==NULL){
return;
}
if(root->val==n){
res[i] = h;
val[i] = root1->val;
return;
}
dfs(root->left,n,h+1,i,root);
dfs(root->right,n,h+1,i,root);
}
bool isCousins(struct TreeNode* root, int x, int y){
dfs(root,x,1,0,root);
dfs(root,y,1,1,root);
if(res[0]==res[1] && res[0]!=0 && res[1]!=0 && val[1]!=val[0]){
return 1;
}
return 0;
}