1. 程式人生 > >B樹的C++程式碼實現

B樹的C++程式碼實現

//這是我在學資料庫時寫的C++的B樹的實現.

B樹有四個性質:

1.樹中每個節點最多含有2m-1的節點;

2.除了根節點外,其他每個節點至少有m-1個孩子;

3.若根節點不是葉子節點則至少有2個孩子(即整個樹只有根節點);

4.所有葉子節點都在同一層

#include <iostream>
using namespace std;
static const int m = 3;  //定義最小度為3
static const int key_max = 2*m-1;//關鍵字最大個數
static const int key_min = m - 1;//關鍵字最小個數
static const int child_max = key_max + 1;//子節點最大個數(這裡可以看出子節點數與關鍵字數有關)
static const int child_min = key_min + 1;//子節點最小個數

template <class T>
class BTree;//前置宣告
/*
類名:BTnode
*/
template <class T>
class BTnode
{
  friend class BTree<T>;//友元函式
public:
	BTnode()
	{
		keyNum = 0;
		parent = NULL;//父節點初始化
        isleaf = true;
		int i ;
		for (i = 0;i < child_max;i++)//子樹指標陣列初始化
		{
			pchild[i] = NULL;
		}
		for (i = 0;i < key_max;i++)//關鍵字陣列初始化
		{
			keyvalue[i] = '\0';
		}
	}
 private:
 bool isleaf;//判斷節點是否是葉子節點
 int keyNum;//關鍵字個數
 BTnode<T>* parent;//指向父節點
 BTnode<T>* pchild[child_max];//子樹指標陣列
 T   keyvalue[key_max];//關鍵字陣列
};

/*
類名:BTree
*/
template <class T>
class BTree
{
public:
	  /*
     函式名:BTree
     函式作用:無參建構函式
     函式引數:無
     函式返回值: 無
     */
  
    BTree()
    {
       root = NULL;
    }
      /*
     函式名:BTree_insert
     函式作用:插入關鍵字
     函式引數:T value
     函式返回值: bool型別判斷插入是否成功
     */
 
    bool BTree_insert(T value)
    {
      if (contain(value))//看樹中是否有相同的關鍵字
      {
         return false;
      }
      else
      {
      	  if (root == NULL)
      	  {
      	  	 root = new BTnode<T>();
      	  }
      	  if (root->keyNum == key_max)
      	  {
             BTnode<T>* newnode = new BTnode<T>();
             newnode->pchild[0] = root;
             newnode->isleaf = false;//由上步操作可知newnode已經有子節點
             SplitBlock(newnode,0,root);//分塊操作
             root = newnode;//更新根節點
      	  }
      	  Notfull_insert(root,value);//插入塊未滿的操作
      	  return true;
      }
    }

    /*
     函式名:SplitBlock
     函式作用:把節點分裂
     函式引數:BTnode<T>* node_parent,int child_index,BTnode<T>* node_child
     函式返回值: 無
     */
    
    void SplitBlock(BTnode<T>* node_parent,int child_index,BTnode<T>* node_child)
    {
       BTnode<T>* node_right = new BTnode<T>();
       node_right->isleaf = node_child->isleaf;
       node_right->keyNum = key_min;
       int i;
       //node_right拷貝關鍵字
       for (i = 0;i < key_min;i++)
       {
         node_right->keyvalue[i] = node_child->keyvalue[i+child_min];
       }
       //判斷分裂的節點是否是葉子節點,如果不是的話就要把它的孩子賦值過去
       if (!node_child->isleaf)
       {
          for (i = 0;i < child_min;i++)
          {
             node_right->pchild[i] = node_child->pchild[i+child_min];
             node_child->pchild[i+child_min]->parent = node_right->pchild[i];
          }
       }
       node_child->keyNum = key_min;//更新子節點的關鍵字數
       //把父節點關鍵字和子指標往後移 倒序賦值
       for (i = node_parent->keyNum;i > child_index;i--)
       {
          node_parent->keyvalue[i] = node_parent->keyvalue[i-1];
          node_parent->pchild[i+1] = node_parent->pchild[i];
          node_child->pchild[i]->parent = node_parent->pchild[i+1];
       }
       node_parent->keyNum++;//更新父節點關鍵字數
       node_parent->pchild[child_index+1] = node_right;
       node_right->parent = node_parent->pchild[child_index+1];
       //把中間的那個關鍵字上移到父節點
       node_parent->keyvalue[child_index] = node_child->keyvalue[key_min];
    }

   
    /*
    函式名:Notfull_insert
    函式作用:往沒有滿的節點中增加關鍵字
    函式引數:BTnode<T>* node,T value
    函式返回值:無
    */
    void Notfull_insert(BTnode<T>* node,T value)
    {

      int node_keynum = node->keyNum;//獲取節點關鍵字數
      if (node->isleaf)//node是葉子節點
      {
      	 while (node_keynum > 0 && value < node->keyvalue[node_keynum-1])
      	 {
      	 	node->keyvalue[node_keynum] = node->keyvalue[node_keynum-1];//把關鍵字往後移動
      	 	--node_keynum;
      	 }
      	 node->keyvalue[node_keynum] = value;
      	 node->keyNum++;
      }
      else//node是內部節點
      {
      	  while (node_keynum > 0 && value < node->keyvalue[node_keynum-1])
      	  {
      	  	 --node_keynum;
      	  }
      	  //在比它小和比它大中間那個子節點中找合適位置,
      	  //如果它比所有的都小則在第一個子節點中尋找
      	  BTnode<T>* node_child = node->pchild[node_keynum];
      	  if (node_child->keyNum == key_max)
      	  {
      	  	 SplitBlock(node,node_keynum,node_child);
      	  	 if (value > node->keyvalue[node_keynum])//插入值和子節點中間的值比較
      	  	 {
      	  	 	 node_child = node->pchild[node_keynum+1];
      	  	 }
      	  }
      	  Notfull_insert(node_child,value);//繼續往下尋找合適的位置
      }
    }
    
    /*
    函式名:contain
    函式作用:查詢是否有相同元素在數中
    函式引數:T value
    函式返回值:bool型別
    */
   bool contain(T value)
   {
        if (BTree_find(root,value) != NULL)
   	  return true;
    return false;
   }
   
    /*
    函式名:BTree_find
    函式作用:查詢是否有相同元素在數中
    函式引數:BTnode<T>* node,T value
    函式返回值: BTnode<T>*
    */
   BTnode<T>* BTree_find(BTnode<T>* node,T value)
   {
   	   if (node == NULL)//當前節點為空的時候
   	   {
   	   	  return NULL;
   	   }
   	   else//當前節點不為空
   	   {
   	   	 int i;
         //在比它小和比它大的中間子節點中尋找相等的
   	   	 for (i = 0; i < node->keyNum ;i++)
   	   	 {
   	   	 	  if (value <= node->keyvalue[i])
            {
               break; 
            }
   	   	 }

   	   	 //校驗當前的關鍵字是否等於查詢的關鍵字
         if (i < node->keyNum && value == node->keyvalue[i])//i是下標最大值keyNum-1
         {
            return node;
         }
         else
         {
         	//如果之前比查詢關鍵沒有相等的關鍵字並且當前節點是葉子節點的話
         	//那麼在B樹中沒有一樣的關鍵字(因為後面比關鍵字更大)
         	 if (node->isleaf)
         	 {
         	 	return NULL;
         	 }
         	 else//如果不是葉子節點則在下面的子節點中尋找
         	 {
             return BTree_find(node->pchild[i],value);//這裡的return別忘了
         	 }
         }
   	   }
   }
  
    /*
    函式名:printpoint
    函式作用:列印元素
    函式引數:BTnode<T>* node,int count
    函式返回值:無
    */
   void printpoint(BTnode<T>* node,int count)
   {
   	  if (node != NULL)//判斷元素是否為空
   	  {
   	  	  int i,j;
          //每個節點從小到大列印
   	     for (i = 0;i < node->keyNum; i++)
   	     {
          //判斷是否葉節點,不是的話則要往子節點中尋找列印
   	   	   if (!node->isleaf)//判斷是否是葉子節點
   	   	   {
             printpoint(node->pchild[i],count-3);
   	   	   }
   	   	   for (j = count;j >= 0; j--)
   	   	   {
   	   	   	 cout<<"-";
   	   	   }
   	   	   cout<<"|"<<node->keyvalue[i]<<"|"<<endl;
   	    }
   	    if (!node->isleaf)//子節點數比關鍵字數多一個
   	   	{
             printpoint(node->pchild[i],count-3);
   	   	}
   	  }
   }
    /*
    函式名:printpoint
    函式作用:printpoint無參函式傳遞值過去
    函式引數:無
    函式返回值:無
    */
   void printpoint()
   {
   	 printpoint(root,key_max*5);
   }
    /*
    函式名:BTree_delete
    函式作用:刪除函式
    函式引數:T value
    函式返回值:bool型別
    */
   bool BTree_delete(T value)
   {
      if (!contain(value))
      {
          return false;
      }
      if (root->keyNum == 1)
      {
             if (root->isleaf)
             {
                delete root;
                root = NULL;
                return true;
             }
             else//當根節點只有一個關鍵字時且不為葉子節點
             {
                BTnode<T>* node_child1 = root->pchild[0];
                BTnode<T>* node_child2 = root->pchild[1];
                //減少一層樹的高度
                if (node_child1->keyNum == key_min && node_child2->keyNum == key_min)
                {
                    MergeBlock(root,0);
                    delete root;
                    root = node_child1;
                }
             }
      }
      BTree_deletebalance(root,value);
   }
     /*
    函式名:MergeBlock
    函式作用:合併塊
    函式引數:BTnode<T>* node_parent,int child_index
    函式返回值:無
    */
   void MergeBlock(BTnode<T>* node_parent,int child_index)
   {
      BTnode<T>* node_child1 = node_parent->pchild[child_index];
      BTnode<T>* node_child2 = node_parent->pchild[child_index+1];
      //將父節點的節點對應的關鍵字下移
      node_child1->keyvalue[key_min] = node_parent->keyvalue[child_index];
      node_child1->keyNum = key_max;
      int i;
      for (i = 0;i < key_min;i++)
      {
        node_child1->keyvalue[child_min+i] = node_child2->keyvalue[i];
      }
      //判斷node_child1是否是葉子節點,不是葉子節點則要將node_child2的子節點賦值給node_child1
      if (!node_child1->isleaf)
      {
        for (i = 0; i < child_min;i++)
        {
           node_child1->pchild[i+child_min] = node_child2->pchild[i];
        }
      }
      //因為父節點下移一個關鍵字,則關鍵字後的所有值要往前移動一個
      node_parent->keyNum--;
      for (i = child_index;i <  node_parent->keyNum;i++)
      {
         node_parent->keyvalue[i] = node_parent->keyvalue[i+1];
         //子節點也要往前移動一位
         node_parent->pchild[i+1] = node_parent->pchild[i+2];
      }
      delete node_child2;
      node_child2 = NULL;
   }
      /*
    函式名:getprev
    函式作用:在左邊的兄弟節點中找一個最大的
    函式引數:BTnode<T>* node
    函式返回值:左邊兄弟的最大的關鍵字
    */
   T getprev(BTnode<T>* node)
   {
    //在比節點位置小的節點中找一個最大的值最為
      while (!node->isleaf)
      {
         node = node->pchild[node->keyNum];
      }
      node->keyNum--;
      return node->keyvalue[node->keyNum-1];
   }
      /*
    函式名:getnext
    函式作用:在右邊的兄弟節點中找一個最小的
    函式引數:BTnode<T>* node
    函式返回值:右邊兄弟的最小的關鍵字
    */
   T getnext(BTnode<T>* node)
   {
     //在比節點位置大的節點中找一個最小的
     while (!node->isleaf)
      {
         node = node->pchild[0];
      }
     return node->keyvalue[0];
   }
    /*
    函式名:BTree_deletebalance
    函式作用:用遞迴刪除並做修復
    函式引數:BTnode<T>* node
    函式返回值:無
    */
   void BTree_deletebalance(BTnode<T>* node,T value)
   {
     int i;
     //在當前節點中找合適座標使得value在這個區間內
     for(i = 0;i < node->keyNum && value > node->keyvalue[i];i++)
     {

     }
     //如果當前i的關鍵字等於value
     if (i < node->keyNum && value == node->keyvalue[i])
     {
      //判斷當前節點是否是葉子節點,如果是的話直接刪除,下面的情況保證瞭如果value在葉子節點的話,葉子節點keyNum一定是>=child_min
        if (node->isleaf)
        {
           node->keyNum--;
           //把i後面的都往前移動一位
           for (;i < node->keyNum;i++)
           {
             node->keyvalue[i] = node->keyvalue[i+1];
           }
           return;
        }
        else//當前節點為內節點
        {
          //裡面關鍵字都比value小的節點
          BTnode<T>* node_left = node->pchild[i];
          //裡面關鍵字都比value大的節點
          BTnode<T>* node_right = node->pchild[i+1];
          if (node_left->keyNum >= child_min)//左子節點中的關鍵字數大於等於child_min
          {
             T prev_replace = getprev(node_left);
             //傳遞不平衡點讓pre_replace
             BTree_deletebalance(node_left,prev_replace);
             //讓前繼的關鍵字替換當前刪除的關鍵字
             node->keyvalue[i] = prev_replace;
             return;
          }
          else if (node_right->keyNum >= child_min)//右子節點中的關鍵字數大於等於child_min
          {
             T next_replace = getnext(node_right);
             //在左邊中找到最大的那個遞迴替換
             BTree_deletebalance(node_right,next_replace);
             //讓前繼的關鍵字替換當前刪除的關鍵字
             node->keyvalue[i] = next_replace;
             return;
          }
          else//左右子節點中的關鍵字數都等於key_min
          {
            //合併兩個子節點
            MergeBlock(node,i);
            //在合併的節點中刪除value值
            BTree_deletebalance(node_left,value);
          }
        }
     }
     else//關鍵字不在當前節點(下面步驟保證了遍歷的所有節點關鍵字數都是大於等於child_min)
     {
       //在(<<value<<)的區間找
        BTnode<T>* node_child = node->pchild[i];//這裡i = node->keyNum
        BTnode<T>*  node_left = NULL;
        BTnode<T>*  node_right = NULL;
        if (node_child->keyNum == key_min)//當前節點只有兩個關鍵字,補給當前節點使之大於等於child_min
        {  
            //判斷是否有左右兄弟節點
            if (i >0)
            {
                node_left = node->pchild[i-1];
            }
            if (i <= node->keyNum-1)
            {
               node_right = node->pchild[i+1];
            }
           int j;
           //當前左兄弟的關鍵字數大於等於child_min
           if (node_left && node_left->keyNum >= child_min)
           {   
                //把父節點的i-1對應的關鍵字下移
               for (j = node_child->keyNum;j > 0; j--)
               {
                 node_child->keyvalue[j] = node_child->keyvalue[j-1];
               }
               node_child->keyvalue[0] = node->keyvalue[i-1];
               //如果子節點的左兄弟節點不是葉子節點
               if (!node_left->isleaf)
               {
                //把左兄弟最右邊的子節點指標賦值給node_child
                  for (j = node_child->keyNum+1;j > 0;j--)
                  {
                    node_child->pchild[j-1]->parent = node_child->pchild[j]->parent;
                    node_child->pchild[j] = node_child->pchild[j-1];

                  }
                  node_left->pchild[node_left->keyNum]->parent =  node_child->pchild[0];
                  node_child->pchild[0] = node_left->pchild[node_left->keyNum];
               }
               node_child->keyNum++;
               node->keyvalue[i-1] = node_left->keyvalue[node_left->keyNum-1];
               node_left->keyNum--;
           }
           else if (node_right && node_right->keyNum >= child_min)
           {
               //把父節點的i對應的關鍵字下移
               node_child->keyvalue[node_child->keyNum] = node->keyvalue[i];
               node_child->keyNum++;
               //把右兄弟節點最小的關鍵字賦值給父節點的i位置
               node->keyvalue[i] = node_right->keyvalue[0];
               node_right->keyNum--;
               //把右兄弟的關鍵字往前移動一位
               for (j = 0;j < node_right->keyNum;j++)
               {
                 node_right->keyvalue[j] = node_right->keyvalue[j+1];
               }
               //如果右兄弟不是葉子節點則需要把右兄弟最左邊的子節點指標賦值給node_child
               if (!node_right->isleaf)
               {
                node_right->pchild[0]->parent = node_child->pchild[node_child->keyNum]->parent;
                 node_child->pchild[node_child->keyNum] = node_right->pchild[0];
                  for (j = 0;j < node_right->keyNum+1;j++)
                  {
                    node_right->pchild[j+1]->parent = node_right->pchild[j]->parent;
                    node_right->pchild[j] = node_right->pchild[j+1];

                  }
                 
               }
           }
           else if (node_left)//node_left只有child_min-1個關鍵字
           {
            //把左兄弟合併
             MergeBlock(node,i-1);
             //更新子節點
             node_child = node_left;
           }
           else if (node_right)
           {
            //把右兄弟合併,子節點無需更新
             MergeBlock(node,i);
           }
        }
        BTree_deletebalance(node_child,value);
     }
   }
private:
	BTnode<T>* root;//根節點
};