Question

所以我遇到的问题是AVLtree的插入方法。我的avltree存储城市对象，我使用城市的名称来决定他们应该去哪里。

我的城市.cpp课程：

 #include "City.h"
 #include <iostream>

  using namespace std;              


  City::City(string name, string country, double lat, double lon){
         this->name = name;
         this->country = country;
         this->latitude = lat;
         this->longtitude = lon;
             }

  double City::getLatOfCity(void){
         return this->latitude;
         }

  double City::getLonOfCity(void){
         return this->longtitude;
         }

  string City::getName(void){
         return this->name;
         }

  string City::getCountry(void){
         return this->country;
         }

我的avltree .cpp文件中的方法。

我的左旋转方法：

  Node* AVLtree::rotateLeft(Node* node){
       Node* tempParent= node->getParent();
       Node* tempNode = node->getChildL();
       node->setChildL(tempNode->getChildR());
       tempNode->setChildR(node);

       if(tempParent==0){tempNode->removeParent();this->rootNode=tempNode;}
       else{tempParent->setChildL(tempNode);
            //tempNode->updateParent(tempParent);
            }

       return tempNode;
       }

向右旋转方法：

  Node* AVLtree::rotateRight(Node* node){ 
       Node* tempParent = node->getParent();
       if(tempParent!=0){cout<<"IHN";cout<<"temp parent: " << tempParent->getCity()->getName();}
       Node* tempNode = node->getChildR();
       node->setChildR(tempNode->getChildL());
       tempNode->setChildL(node);

       if(tempParent==0){tempNode->removeParent();this->rootNode=tempNode;}
       else{tempParent->setChildR(tempNode);}

       return tempNode;
       }

我的第一个双右旋转方法。这个不能正常工作，因为我在方法内部传递给旋转方法的指针是不同的。

   /*  
  Node* AVLtree::rotateTwoRights(Node* node){
        node = rotateRight(node->getChildR());
        node = rotateRight(node->getParent());
        return node;
        }    */

我知道这非常混乱，但它似乎正确地旋转了周围的节点。

  Node* AVLtree::rotateTwoRights(Node* node){
       Node* tempParent = node;
       Node* tempNode = node->getChildR();
       Node* tempTempNode = tempNode->getChildL();
       tempNode->setChildL(tempTempNode->getChildL());
       tempTempNode->setChildR(tempNode);
       tempParent->setChildR(tempTempNode);
       Node* tempTempParent = tempParent->getParent();
       tempTempParent->setChildR(tempParent->getChildL());
       tempParent->setChildL(tempTempParent);
       Node* newTempParent = tempParent->getParent();

       if(newTempParent==0){tempParent->removeParent();this->rootNode=tempParent;}
       else{newTempParent->setChildR(tempParent);} 

        return tempParent;
        }

与我的第一个旋转双右方法相同，但是对于左侧。同样的问题：

  Node* AVLtree::rotateTwoLefts(Node* node){
        node = rotateRight(node->getChildL());
        node = rotateLeft(node);
        return node;
        }

我的插入方法：

  Node* AVLtree::insertNode(Node* parent, Node* node, City *city, int side){
       if(node == 0){
               if(side==0){
                     node = parent->setChildL(new Node(city));
               }else if(side==1){
                     node = parent->setChildR(new Node(city));
               } 
       }else if(node->getCity()->getName().compare(city->getName())<0){ //Right case
             parent = node;
             node = insertNode(parent, node->getChildR(), city, 1);
             if(parent->getBalance()==2){
                   if(parent->getChildR()->getCity()->getName().compare(city->getName())<0){
                         node = rotateRight(parent);
                   }else{
                         node = rotateTwoRights(parent);
                   }                                
             } 
       }else if(node->getCity()->getName().compare(city->getName())>0){ //Left case
             parent = node;
             node = insertNode(parent, node->getChildL(), city, 0);
             if(parent->getBalance()==-2){
                   if(parent->getChildL()->getCity()->getName().compare(city->getName())>0){
                         node = rotateLeft(parent);
                   }else{
                         node = rotateTwoLefts(parent);
                   }
             }       
       }else{
             node=0;
       }
       return node;
       }

所以问题在于我尝试插入'CA'。

所以树看起来像这样：

   B
  / \
 A   C
      \
       D

并在插入'CA'后，应遵循以下步骤：

1）

   B
  / \
 A   C
      \
       D
      /
    CA

2）

3）

    C
   / \
  B   CA
 /     \
A       D

当我对它运行测试时，错误是树的根仍然是B. 我相信这是由parent引起的，因为在递归调用后重新平衡树时它不会更新。但是，当我将parent分配给方法的返回值时，我甚至无法将“C”添加到树中。

我会非常感谢任何形式的帮助。我花了很多时间在这上面，真的很想完成这件事。

我为长篇文章和凌乱的代码道歉，并提前感谢！

Answer 1

鉴于节点C是第一个节点，其中左子树的深度（0）和右子树的深度（2）相差2，我认为应该只有一个简单的左旋转，屈服

  B
 / \
A   CA
   / \
  C   D

当不平衡的补丁达到B时，树就足够平衡了。我没有详细查看您的代码，但您的假设结果似乎不正确，即代码实际上可能是。

C ++ AVLtree插入方法

1 个答案: