/* 用于获取x节点的深度
*/
int get_dep(struct TreeNode *node, int x, int dep){
if(!node) return -1;
if(node->val == x){
return dep+1;
}
int l = get_dep(node->left, x, dep);
if(l > 0) return l+1;
int r = get_dep(node->right, x, dep);
if(r > 0) return r+1;
return -1;
}
/*
* 用于检测,是否存在x与y节点为兄弟节点。
*/
bool check(struct TreeNode *node, int x, int y){
if(!node) return false;
if(node->left && node->right &&
node->left->val == x && node->right->val == y )
return true;
if(node->left && node->right &&
node->left->val == y && node->right->val == x )
return true;
if(check(node->left, x, y) || check(node->right, x, y)){
return true;
}
return false;
}
bool isCousins(struct TreeNode* root, int x, int y){
if(!root) return false;
int x_d = get_dep(root, x, 0);
int y_d = get_dep(root, y, 0);
printf("%d, %d \n", x_d, y_d);
if(x_d == y_d && !check(root, x, y))
return true;
return false;
}
leetcode-993. 二叉树的堂兄弟节点-C语言
最新推荐文章于 2021-05-17 12:43:42 发布