二叉树为:
1
2 3
4 5 6 7
程序如下:
#include<iostream>
#define N 7
using namespace std;
typedef struct node
{
struct node *leftChild;
struct node *rightChild;
int data;
}BiTreeNode, *BiTree;
BiTreeNode *createNode(int i)
{
BiTreeNode * q = new BiTreeNode;
q->leftChild = NULL;
q->rightChild = NULL;
q->data = i;
return q;
}
BiTree creatBiTree()
{
BiTreeNode *p[N] = {NULL};
int i;
for(i = 0; i < N; i++)
p[i] = createNode(i + 1);
for(i = 0; i < N/2; i++)
{
p[i]->leftChild = p[i * 2 + 1];
p[i]->rightChild = p[i * 2 + 2];
}
return p[0];
}
int max(int a, int b )
{
return a < b ? b : a;
}
// 用递归方法求树的高度
int Depth(BiTree T)
{
if (NULL == T)
return 0;
int leftDepth = Depth(T->leftChild);
int rightDepth = Depth(T->rightChild);
return 1+ max(leftDepth, rightDepth);
}
void PrintNodeAtLevel(BiTree T,int level)
{
// 空树或层级不合理
if (NULL == T || level < 1 )
return;
if (1 == level)
{
cout << T->data << " ";
return;
}
// 左子树的 level - 1 级
PrintNodeAtLevel(T->leftChild, level - 1);
// 右子树的 level - 1 级
PrintNodeAtLevel(T->rightChild, level - 1);
}
void LevelTraverse(BiTree T)
{
if (NULL == T)
return;
int depth = Depth(T);
int i;
for (i = 1; i <= depth; i++)
{
PrintNodeAtLevel(T, i);
cout << endl;
}
}
int main()
{
BiTree T = creatBiTree();
cout << "层次遍历:" << endl;
LevelTraverse(T);
cout << endl;
return 0;
}
结果为:
层次遍历:
1
2 3
4 5 6 7