leetcode-993. 二叉树的堂兄弟节点-C语言


/* 用于获取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;
}


猜你喜欢

转载自blog.csdn.net/weixin_36094222/article/details/92799353