二叉搜索树 getHeight() 函数

Binary Search Tree getHeight() function

本文关键字:函数 getHeight 搜索树      更新时间:2023-10-16

我正在尝试编写一个将返回BST高度的函数。 我正在使用测试驱动开发编写代码,但不幸的是,它没有通过我的测试,我无法弄清楚为什么。

我通过了测试的第 0、1 和 2 部分,但在第 3 部分失败了。 非常感谢任何帮助找出我的代码出了什么问题。 谢谢!

以下是我的代码的一些相关摘录:

英国夏令时.cpp

unsigned BST::Node::getHeight() const{
    unsigned leftSide = 0;
    if(myLeft != NULL){
        myLeft->getHeight();
    }
    unsigned rightSide = 0;
    if(myRight != NULL){
        myRight->getHeight();
    }
    if(leftSide > rightSide){
        return leftSide++;
    } else{
        return rightSide++;
    }
}
unsigned BST::getHeight() const{
    if(myNumItems == 0){
        return 0;
    } else if(myNumItems == 1){
        return 1;
    } else if(myNumItems == 2){
        return 2;
    } else {
        return myRoot->getHeight();
    }
}

BST.h

class BST {
public:
    BST();
    virtual ~BST();
    unsigned getHeight() const;
private:
struct Node {
    Node(const Item& it);
        virtual ~Node();
        void insert(const Item& it);
        unsigned getHeight() const;
        Item myItem;
        Node* myLeft;
        Node* myRight;
};
Node* myRoot;
unsigned myNumItems;
friend class BST_Tester;
};

BST_Tester.cpp

void BST_Tester::testGetHeight() {
cout << "Testing getHeight()..." << flush;
BST bst;
// empty
assert( bst.getHeight() == 0 );
cout << " 0 " << flush;
// balanced
bst.insert(44);
assert( bst.getHeight() == 1 );
cout << " 1 " << flush;
bst.insert(22);
assert( bst.getHeight() == 2 );
cout << " 2 " << flush;
bst.insert(66);
assert( bst.getHeight() == 2 );
cout << " 3 " << flush;
bst.insert(77);
assert( bst.getHeight() == 3 );
cout << " 4a " << flush;
bst.insert(55);
assert( bst.getHeight() == 3 );
cout << " 4b " << flush;
bst.insert(33);
assert( bst.getHeight() == 3 );
cout << " 4c " << flush;
bst.insert(11);
assert( bst.getHeight() == 3 );
cout << " 4d " << flush;
bst.insert(88);
assert( bst.getHeight() == 4 );
cout << " 4e " << flush;
// chained ascending
  BST bst2a;
  bst2a.insert(11);
  bst2a.insert(22);
  bst2a.insert(33);
  bst2a.insert(44);
  bst2a.insert(55);
  bst2a.insert(66);
  bst2a.insert(77);
  assert( bst2a.getHeight() == 7 );
  cout << " 5a " << flush;
// chained descending
  BST bst2b;
  bst2b.insert(77);
  bst2b.insert(66);
  bst2b.insert(55);
  bst2b.insert(44);
  bst2b.insert(33);
  bst2b.insert(22);
  bst2b.insert(11);
  assert( bst2b.getHeight() == 7 );
  cout << " 5b " << flush;
// four 4-node permutations
BST bst3;
bst3.insert(44);
assert( bst3.getHeight() == 1 );
  cout << " 6a " << flush;
bst3.insert(22);
assert( bst3.getHeight() == 2 );
  cout << " 6b " << flush;
bst3.insert(33);
assert( bst3.getHeight() == 3 );
  cout << " 6c " << flush;
bst3.insert(55);
assert( bst3.getHeight() == 3 );
  cout << " 6d " << flush;
BST bst4;
bst4.insert(44);
assert( bst4.getHeight() == 1 );
  cout << " 7a " << flush;
bst4.insert(33);
assert( bst4.getHeight() == 2 );
  cout << " 7b " << flush;
bst4.insert(22);
assert( bst4.getHeight() == 3 );
  cout << " 7c " << flush;
bst4.insert(55);
assert( bst4.getHeight() == 3 );
  cout << " 7d " << flush;
BST bst5;
bst5.insert(44);
assert( bst5.getHeight() == 1 );
  cout << " 8a " << flush;
bst5.insert(33);
assert( bst5.getHeight() == 2 );
  cout << " 8b " << flush;
bst5.insert(55);
assert( bst5.getHeight() == 2 );
  cout << " 8c " << flush;
bst5.insert(66);
assert( bst5.getHeight() == 3 );
  cout << " 8d " << flush;
BST bst6;
bst6.insert(44);
assert( bst6.getHeight() == 1 );
  cout << " 9a " << flush;
bst6.insert(33);
assert( bst6.getHeight() == 2 );
  cout << " 9b " << flush;
bst6.insert(66);
assert( bst6.getHeight() == 2 );
  cout << " 9c " << flush;
bst6.insert(55);
assert( bst6.getHeight() == 3 );
  cout << " 9d " << flush;
cout << " Passed!" << endl;
}

编辑:

如果有人看到这个并想知道答案,我想通了。 这是我的新代码。

unsigned BST::Node::getHeight() const{
    unsigned leftSide = 0;
    if(myLeft != NULL){
        leftSide = myLeft->getHeight();
    }
    unsigned rightSide = 0;
    if(myRight != NULL){
        rightSide = myRight->getHeight();
    }
    if(leftSide > rightSide){
        return ++leftSide;
    } else{
        return ++rightSide;
    }
}
unsigned BST::getHeight() const{
    if(myNumItems == 0){
        return 0;
    } else {
        return myRoot->getHeight();
    }
}

这只是将 leftSide++ 翻转到 ++leftSide 的问题。

丢弃你的代码。试着这样做,伙计

int getHeight(tree *p)
{
   if (!p)
       return 0;
   int left = getHeight(p->left);
   int right = getHeight(p->right); 
   return max(left, right) +1;
}