#include<iostream.h>

#include"B_Tree.h"

void main()
{
	KeyType key;       //用于保存記錄的關鍵字
	MBNode *mt;        //定義指向B_樹結點的指針
	InitMBTree(mt);    //初始化mt
	bool Tag;
	while(1){
		cout<<"功能菜單："<<endl;
		cout<<'\t'<<"1.從鍵盤輸入一組關鍵字插入B_樹"<<endl;
		cout<<'\t'<<"2.向B_樹插入一個關鍵字"<<endl;
		cout<<'\t'<<"3.從B_樹中刪除一個關鍵字"<<endl;
		cout<<'\t'<<"4.從B_樹中刪除一組關鍵字"<<endl;
		cout<<'\t'<<"5.遍歷輸出B_樹中所有關鍵字"<<endl;
		cout<<'\t'<<"6.顯示一棵樹"<<endl;
		cout<<'\t'<<"7.清除B_樹，即回收B_樹中的所有結點"<<endl;
		cout<<'\t'<<"8.結束程序運行"<<endl;
		cout<<endl;
		int m;
		cout<<"輸入你的選擇（1-8）：";
		cin>>m;
		switch(m){
		case 1:
			cout<<endl;
			cout<<"從鍵盤上輸入一組關鍵字的個數：";
			int i,n;cin>>n;
			cout<<"從鍵盤上輸入"<<n<<"個關鍵字（以回車為界）："<<endl;
			for(i=0;i<n;i++){
				cin>>key;
				InsertMBTree(mt,key);
			}
			cout<<endl;
			DisplayMBTree(mt);
			cout<<endl;
			break;
		case 2:
			cout<<endl;
			cout<<"從鍵盤上輸入一個待插入的關鍵字：";
			cin>>key;
			InsertMBTree(mt,key);
			cout<<endl;
			DisplayMBTree(mt);
			cout<<endl;
			break;
		case 3:
			cout<<endl;
			cout<<"從鍵盤上輸入一條帶刪除記錄的關鍵字：";
			cin>>key;
			Tag=DeleteMBTree(mt,key);
			cout<<endl;
			if(Tag)
				cout<<"刪除成功！"<<endl;
			else cout<<"刪除失??！"<<endl;
			cout<<endl;
			DisplayMBTree(mt);
			cout<<endl;
			break;
		case 4:
			cout<<endl;
			cout<<"從鍵盤上輸入要刪除的關鍵字的個數：";
			int	m;cin>>m;
			cout<<"從鍵盤上輸入"<<m<<"個關鍵字（以回車為界）："<<endl;
			for(i=0;i<m;i++){
				cin>>key;
				Tag=DeleteMBTree(mt,key);
				cout<<endl;
				if(Tag)
					cout<<"刪除成功！"<<endl;
				else cout<<"刪除失??！"<<endl;
				cout<<endl;
			}
			DisplayMBTree(mt);
			break;
		case 5:
			cout<<endl;
			TravelMBTree(mt);
			cout<<endl;
			break;
		case 6:
			cout<<endl;
			DisplayMBTree(mt);
			cout<<endl;
			break;
		case 7:
			ClearMBTree(mt);
			break;
		default:
			ClearMBTree(mt);
			cout<<"運行程序結束，再見！"<<endl<<endl;
			return;
		}
	}
}