#include <iostream.h>
#include <iomanip.h>
#include "AVLTree.h"


void main()
{
	char select;
	cout<<"選擇avl樹結點的類型: i:整數 c:單字符 f:浮點數 q:退出程序"<<endl;
	cin>>select;
	for(;(select!='i'&&select!='c'&&select!='f'&&select!='q');)//排除輸入錯誤
	{
		cout<<"輸入有錯誤"<<endl;
		cin>>select;
	}
	switch (select)
	{	case 'i':
		{
			avlTree<int> treeint;
			int val;char temp;
			cout<<"建立樹:輸入整形結點值，以-1000為結尾"<<endl;
			cin>>val;
			for(;val!=-1000;)
			{
				treeint.add(val);
				cin>>val;
			}
			cout<<"選擇功能: a:增加結點，d:刪除結點，f:查找結點，n:刪除本樹全部結點，v:顯示當前樹，q:退出程序"<<endl;
			do{
				cin>>select;
				for(;(select!='a'&&select!='d'&&select!='f'&&select!='n'&&select!='v'&&select!='q');)
				{
					cout<<"輸入有錯誤"<<endl;
					cin>>select;
				}
				switch (select)
				{
					case 'a':cout<<"輸入插入值:";cin>>val;treeint.add(val);break;
					case 'd':cout<<"輸入刪除值:";cin>>val;treeint.remove(val);break;
					case 'f':cout<<"輸入查找值:";cin>>val;treeint.display(treeint.findValue(val));break;
					case 'n':cout<<"真的刪除全部么(y/n)?";cin>>temp;if(temp=='y') treeint.deleteAllValue();else break;break;
					case 'v':treeint.display();break;
					case 'q':cout<<"Bye";return;
				}
				cout<<"選擇功能: a:增加結點，d:刪除結點，f:查找結點，n:刪除本樹全部結點，v:顯示當前樹，q:退出程序"<<endl;
			}while (select!='q');
			return;
		}
		case 'c':
		{
			avlTree<char> treeint;
			char val;char temp;
			cout<<"建立樹:輸入單個字符結點值，以 ~ 為結尾"<<endl;
			cin>>val;
			for(;val!='~';)
			{
				treeint.add(val);
				cin>>val;
			}
			cout<<"選擇功能: a:增加結點，d:刪除結點，f:查找結點，n:刪除本樹全部結點，v:顯示當前樹，q:退出程序"<<endl;
			do{
				cin>>select;
				for(;(select!='a'&&select!='d'&&select!='f'&&select!='n'&&select!='v'&&select!='q');)
				{
					cout<<"輸入有錯誤"<<endl;
					cin>>select;
				}
				switch (select)
				{
					case 'a':cout<<"輸入插入值:";cin>>val;treeint.add(val);break;
					case 'd':cout<<"輸入刪除值:";cin>>val;treeint.remove(val);break;
					case 'f':cout<<"輸入查找值:";cin>>val;treeint.display(treeint.findValue(val));break;
					case 'n':cout<<"真的刪除全部么(y/n)?";cin>>temp;if(temp=='y') treeint.deleteAllValue();else break;break;
					case 'v':treeint.display();break;
					case 'q':cout<<"Bye";return;
				}
				cout<<"選擇功能: a:增加結點，d:刪除結點，f:查找結點，n:刪除本樹全部結點，v:顯示當前樹，q:退出程序"<<endl;
			}while (select!='q');
			return;
		}
		case 'f':
		{
			avlTree<float> treeint;
			float val;char temp;
			cout<<"建立樹:輸入整形結點值，以-1000為結尾"<<endl;
			cin>>val;
			for(;val!=-1000;)
			{
				treeint.add(val);
				cin>>val;
			}
			cout<<"選擇功能: a:增加結點，d:刪除結點，f:查找結點，n:刪除本樹全部結點，v:顯示當前樹，q:退出程序"<<endl;
			do{
				cin>>select;
				for(;(select!='a'&&select!='d'&&select!='f'&&select!='n'&&select!='v'&&select!='q');)
				{
					cout<<"輸入有錯誤"<<endl;
					cin>>select;
				}
				switch (select)
				{
					case 'a':cout<<"輸入插入值:";cin>>val;treeint.add(val);break;
					case 'd':cout<<"輸入刪除值:";cin>>val;treeint.remove(val);break;
					case 'f':cout<<"輸入查找值:";cin>>val;treeint.display(treeint.findValue(val));break;
					case 'n':cout<<"真的刪除全部么(y/n)?";cin>>temp;if(temp=='y') treeint.deleteAllValue();else break;break;
					case 'v':treeint.display();break;
					case 'q':cout<<"Bye";return;
				}
				cout<<"選擇功能: a:增加結點，d:刪除結點，f:查找結點，n:刪除本樹全部結點，v:顯示當前樹，q:退出程序"<<endl;
			}while (select!='q');
			return;
		}
		case 'q':cout<<"Bye."<<endl;return;
	}
}