0

我正在编写一个程序来查找二叉树（不是 BST）中最近的祖先。我找到了一个示例工作代码：

mynode *closestAncestor(mynode* root, mynode* p, mynode* q)
{
   mynode *l, *r, *tmp;

   if(root == NULL)
   {
      return(NULL);
   }

 if(root->left==p || root->right==p || root->left==q || root->right==q)

   {
     return(root);
   }
   else
   {
      l = closestAncestor(root->left, p, q);
      r = closestAncestor(root->right, p, q);

      if(l!=NULL && r!=NULL)
      {
        return(root);
      }
      else
      {
         tmp = (l!=NULL) ? l : r;
         return(tmp);
      }
   }
}

我正在尝试执行以下操作（仅传递数据值并仅查找祖先的数据值，而不关心其指针）

int closestanc(node * root, int n1, int n2)
{
    int l, r;
    if(root == NULL)
        return -1;
    if(root->right->data == n1 || root->right->data == n2 || root->left->data == n1 ||     root->left->data == n2)
        return root->data;
    else
    {
        l = closestanc(root->left, n1, n2);
        r = closestanc(root->right, n1, n2);
        if(l!= -1 && r!= -1)
            return root->data;
        else
            return (l != -1 ? l : r);
    }
}
4

1 回答 1

1

您需要检查NULLs。改变：

if(root->right->data == n1 || root->right->data == n2 ||
   root->left->data  == n1 || root->left->data  == n2)

至

if ((root->right != NULL && (root->right->data == n1 || root->right->data == n2))
 || (root->left  != NULL && (root->left->data  == n1 || root->left->data  == n2)))

虽然我怀疑你可以用更简单的替换它：

if (root->data == n1 || root->data == n2)

不改变函数将输出的内容（尽管它会改变它的工作方式）。

附加说明：

该功能不太可靠。似乎如果两者都不存在于树中，它仍然会返回一个祖先。为此，我建议返回-2（或另一个未使用的值）而不是root->data;上述检查，以便您可以确定何时都没有找到。

所以：

if (root->data == n1 || root->data == n2)
    return -2;

然后：

  • 如果函数返回-1，则您知道没有找到任何元素。
  • 如果它返回-2，它只找到其中​​一个。
  • 如果它返回其他任何东西，那就是最近的祖先。
于 2013-06-07T15:13:55.237 回答