用递归方法对二叉树进行层次遍历(某公司实习生招聘笔试试题)

      二叉树为:

                         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

 

 

你可能感兴趣的:(用递归方法对二叉树进行层次遍历(某公司实习生招聘笔试试题))