检查一个节点的值是否等于前两个节点的值的树函数

Question

我想创建一个函数，从每个路径（左或右）中的第三个节点开始检查它是​​否等于树中前两个节点的总和。

这是我的代码，但它不能那样工作：

    typedef struct node
    {
        int data;
        struct node *left;
        struct node *right;
    } node;
    
    int goldtree(node *root)
    {
        int sumleft=0,sumright=0;
        if (root->left==NULL || root->right ==NULL)
            return 0;
        
        sumleft+=root->data+goldtree(root->left);
        sumright+=root->data+goldtree(root->right);
        if (goldtree(root->right)==sumright || sumleft==goldtree(root->left))
            return 1;
    }

这是一个例子：

[ ]

Answer 1

好吧，对不起，我第一次以错误的方式理解了您的问题！ 删除了上一个答案：/

所以，你想检查这棵树是否是golden ，如果我没有错，规则如下：

• 每个当前节点都是前两个节点的总和，除了根节点。
• 如果存在这样一条从根到叶节点的路径，那么这棵树就是金树。

您可以更改您的功能，如下所示：

int goldtree(node *root, int flag){
    int result = 0;
    if(root == NULL) return 0;
  
    /* we have come up to leave node */
    else if(root->left == NULL && root->right == NULL) return 1;

    else{
        /* we would traverse only the path where child node's value - current node's value is equal to prev node' value */            
        if(root->left != NULL && root->left->data - root->data == flag) {
            result = goldtree(root->left, root->data);
        }

        if(root->right != NULL && root->right->data - root->data == flag){
            result = goldtree(root->right, root->data);
        }
        return result;
     }
  
}

调用goldtree驱动函数是：

    goldtree(root, 0);

注意：为什么我们在遍历路径之前检查节点的值是，我们只会考虑那些子节点的值等于当前节点+上一个节点值的路径；

如果您的路径如下所示：

      1
     /
    2
   /
  3

当您在节点2您可以考虑子节点3 prev 两个节点值：

3 - 2 == 1 --> true

Answer 2

您的函数goldtree()以自上而下的方式实现，尝试向下查看树，以便在给定节点上运行时，检查该节点的孙子节点的值。 这是解决问题的可行方法，但您需要处理几个问题，其中包括

• 每个节点最多有四个孙子节点，您需要检查所有子节点，并且
• 您需要准备好处理少于两个子节点和/或少于四个孙节点的节点。

关于您的特定代码，您似乎对goldtree()返回值的重要性感到困惑。 计算诸如root->data+goldtree(root->left)总和没有意义，因为goldtree()返回一个标志（据称）表示以其参数节点为根的子树是否满足条件。 只有当它返回节点的值时，总和才有意义。 然而，即便如此，您也需要为当前节点的每个子节点最多考虑两个节点值总和，并且您不能轻易地用单个int表示它。

同样，您可以解决这些问题以使您的功能成功运行。 但是，我认为以自下而上的方式重新实现它会更清晰，以便您到达的每个节点根据其父节点和祖父节点的值检查自身。 这样做的复杂之处在于您的节点结构没有父指针，因此您需要通过函数参数传达该信息。 通过添加辅助函数，您可以在不更改goldtree()签名的情况下做到这一点：

int goldtree(node *root) {
    return goldtree_helper(root, NULL, NULL);
}

int goldtree_helper(node *current, node *parent, node *grandparent) {
    /*
     * Left as an exercise:
     *
     * behavior depends in part on whether 'current', 'parent' and / or
     * 'grandparent' is NULL.
     *
     * If this node itself passes the test, then the return value also depends
     * on whether its children do.
     */

     /*
      * Example child check, demonstrating cycling the arguments
      * to convey the correct parent and grandparent:
      */
     int left_is_golden = goldtree_helper(current->left, current, parent);
}