如何比较二叉树中的根到叶路径

Question

我试图搜索给定红黑树中所有根到叶的路径。特别是，我想编写一个测试，在给定rbt的情况下，该测试将断言每个路径具有相同数量的黑色节点。

我用两个全局变量尝试这样的东西：

static int count = 0, path = -1;

int check_paths(tree_node t)
{
    if (t == NULL)
    {
        if (path == -1)
            path = count;
        else
            return (path == count);

        return 1;
    }

    if (t->black == 1) ++count;
    int x,y;
    x = check_paths(t->left);
    if (t->black == 1) --count;
    y = check_paths(t->right);

    return (x&&y);
}

然而，当左分支中的黑色节点右侧有红色节点时，我遇到了麻烦，因为这意味着计数比应该减少的更多。

有没有更好的方法来搜索根到叶的路径，计算特定值的频率，然后以某种方式比较计数？或者，如果给定rbt余额，是否有一种完全不同的方法来测试rbt余额（即，它已经制作好，但其正确性不确定）？

Answer 1

考虑如何使函数采用（子）树，并且（a）在平衡时返回其黑色高度，或者（b）在其他情况下返回负数。

基本情况是一个空（子）树；它只返回0。

归纳情况：

• lh是对左子树递归调用的结果

如果lh==lr并且它们是正的，则返回lh 1如果是黑色的

以下是一些未经测试的代码：

int check_paths (tree_node t)
{
    if (t == NULL)
    {
        return 0;
    }
    int lh = check_paths(t->left);
    int rh = check_paths(t->right);

    if (lh != rh || lh < 0)
    {
        return -1;
    }
    else if (t->black == 1)
    {
        return (lh + 1);
    }
    else
    {
        return (lh)
    }
}