使用STL容器構成一棵樹
#include<vector> //vector容器
#include<iostream> //輸入輸出
using namespace std;
const int LEAVES_MAX = 4; //最大子葉數
//樹節點結構
struct TreeNode
{
int num;
vector<TreeNode> Sub;
};
vector<TreeNode>::iterator p; //根節點指標
//用遞迴法生成一棵樹
void CreateTree(vector<TreeNode>::iterator Root)
{
//如果節點值小於等於1,返回
if(Root->num<=1) return;
//新節點指標
TreeNode *newNode;
//為新子葉設定值
for(int i=1;i<=LEAVES_MAX;i++)
{
newNode =new(TreeNode);
newNode->num = Root->num - 1;
Root->Sub.push_back(*newNode);
delete newNode;
}
//遞迴產生子樹
vector<TreeNode>::iterator child;
for(child=Root->Sub.begin();
child!=Root->Sub.end();
child++)
{
CreateTree(child);
}
}
//遞迴顯示樹的內容
void DisplayTree(vector<TreeNode>::iterator Root)
{
//顯示節點內容
cout<<" "<<Root->num;
//如果此節點沒有子葉,返回
if(Root->Sub.empty()==true) return;
//遞迴處理子樹
vector<TreeNode>::iterator child;
for(child=Root->Sub.begin();
child!=Root->Sub.end();
child++)
{
DisplayTree(child);
}
}
void main()
{
TreeNode *Root;
Root = new(TreeNode);
Root->num = 3;
CreateTree(Root);
DisplayTree(Root);
delete Root;
}