编写一个函数,该函数可以有效地使用时间,如果给定的二进制搜索树包含给定的值,则返回1,否则返回0。
例如,对于以下树:
对contains(&n2,3)的调用应返回1,因为根为n2的树包含数字3。
该函数应该返回1,但是它返回0或什么也不返回。
#include <stdlib.h>
#include <stdio.h>
typedef struct Node
{
int value;
struct Node *left;
struct Node *right;
} Node;
int contains(const Node *root, int value)
{
if (root->value == value)
return 1;
else if (value < root->value)
{
if (root->left == NULL)
return 0;
return contains(root->left, value);
}
else if (value > root->value)
{
if (root->right == NULL)
return 0;
return contains(root->left, value);
}
}
int main()
{
Node n1 = {.value=1, .left=NULL, .right=NULL};
Node n3 = {.value=3, .left=NULL, .right=NULL};
Node n2 = {.value=2, .left=&n1, .right=&n3};
printf("%d", contains(&n2, 3));
}
答案 0 :(得分:1)
在root->right, value
时您不返回value > root->value
吗?
else if (value < root->value)
{
if (root->left == NULL)
return 0;
return contains(root->left, value);
}
else if (value > root->value)
{
if (root->right == NULL)
return 0;
return contains(root->right, value); //You need to go to the right branch
}
答案 1 :(得分:1)
在else条件下,您将不转到右侧分支,而转到左侧分支。
else if (value > root->value)
{
if (root->right == NULL)
return 0;
return contains(root->left, value);
}
在值较大的情况下使用下面的行来修复
return contains(root->right, value);