`std::set`

Question

我编写了以下函数来搜索存储整数值的二叉树中的值（该函数是较大程序的一部分）：

bool tree::search(int num)       //the function belongs to class 'tree'
{
   node *temp=head;      //'head' is pointer to root node

   while(temp!=NULL)
   {
      if(temp->data==num)
         break;

      if(num>temp->data)
         temp=temp->right;

      if(num<temp->data)
         temp=temp->left;
   }

   if(temp==NULL)
      return false;
   else if(temp->data==num)
         return true;   
}

问题是：当我搜索树中存在的值时，它运行正常。但是如果我搜索树中不存在的值，程序就会挂起，我必须关闭它。还有一件事 - 我知道我们可以通过传递node * temp作为参数来递归地实现搜索功能，而不是在里面声明它，我已经这样做了，这导致程序正确运行，但我想知道问题是什么在上面的代码中。

我在这里给出了完整的程序，以防它更容易找到错误（请注意我只编写了两个函数）：

#include<iostream>
using namespace std;

struct node
{
int data;
node *left;
node *right;
};

class tree
{
public:
    node *head;    //pointer to root
    int count;     //stores number of elements in tree
    tree();
    void addnode(int);
    void deletenode(int);
    bool search(int);
    int minimum();
    int maximum();
    void inorder();
    void preorder();
    void postorder();
    void printtree();
    int mthlargest();     //finds 'm'th largest element
    int mthsmallest();    //finds 'm'th smallest element
    void convert();       //converts binary tree to linked list
};

tree::tree()
{
   head=NULL;
   count =0;
}

void tree::addnode(int num)
{
   node *temp= new node;
   temp->data=num;
   temp->left=NULL;
   temp->right=NULL;

   node **ptr=&head;          //double pointer

   while(*ptr!=NULL)
   {
      if(num>(*ptr)->data)
         ptr=&((*ptr)->right);

      if(num<(*ptr)->data)
         ptr=&((*ptr)->left);
   }

   *ptr=temp;
}


bool tree::search(int num)
{
   node *temp=head;

   while(temp!=NULL)
   {
      if(temp->data==num)
         break;

      if(num>temp->data)
         temp=temp->right;

      if(num<temp->data)
         temp=temp->left;
   }

   if(temp==NULL)
      return false;
   else if(temp->data==num)
      return true;   
}    




int main()
{
   tree ob;
   ob.addnode(2);

   ob.search(2);

   ob.search(3);

   ob.search(-1);
   ob.search(2);
   cout<<endl<<endl;

   system("pause");
   return 0;
}

旁注：我正在使用Dev C ++编译器和Windows 7操作系统。

Answer 1

放一个else，您的问题就会消失。

因为在temp = temp->right;之后您必须再次检查temp，但在原始代码中，您会立即测试可能不是有效指针的temp->data。

bool tree::search(int num)
{
    node *temp = head;

    while (temp != NULL)
    {
        if (temp->data == num)
            break;

        if (num > temp->data)
            temp = temp->right;
        else                  //  <--- Put this 'else' here
        if (num < temp->data)
            temp = temp->left;
    }

    if (temp == NULL)
        return false;

    if (temp->data == num)
        return true;

    return false;
}

Answer 2

`std::set`

使用std::set;它基本上是STL的二叉树。如果您想搜索某些内容，可以使用count，find或lower_bound。

实现基本数据结构是很好的练习，但在生产中，首先尝试使用STL，因为它们是由具有相关编译器/平台专业知识的专业人员实现的。 Boost是另一组很好的数据结构和常用习语。

在二叉树中搜索

2 个答案:

`std::set`