面试题23:从上往下打印二叉树

#include<iostream>
#include<queue>
using namespace std;
struct Tree
{
	int nValue;
	Tree *pLeft;
	Tree *pRight;
};

void CreateTree(Tree** tree)
{
	(*tree) = (Tree*)malloc(sizeof(Tree));
	(*tree)->nValue = 1;

	(*tree)->pLeft = (Tree*)malloc(sizeof(Tree));
	(*tree)->pLeft->nValue = 2;

	(*tree)->pLeft->pLeft = (Tree*)malloc(sizeof(Tree));
	(*tree)->pLeft->pLeft->nValue = 4;
	(*tree)->pLeft->pLeft->pLeft = NULL;
	(*tree)->pLeft->pLeft->pRight = NULL;

	(*tree)->pLeft->pRight = (Tree*)malloc(sizeof(Tree));
	(*tree)->pLeft->pRight->nValue = 5;
	(*tree)->pLeft->pRight->pLeft = NULL;
	(*tree)->pLeft->pRight->pRight = NULL;



	(*tree)->pRight = (Tree*)malloc(sizeof(Tree));
	(*tree)->pRight->nValue = 3;

	(*tree)->pRight->pLeft = (Tree*)malloc(sizeof(Tree));
	(*tree)->pRight->pLeft->nValue = 6;
	(*tree)->pRight->pLeft->pLeft = NULL;
	(*tree)->pRight->pLeft->pRight = NULL;

	(*tree)->pRight->pRight = (Tree*)malloc(sizeof(Tree));
	(*tree)->pRight->pRight->nValue = 7;
	(*tree)->pRight->pRight->pLeft = NULL;
	(*tree)->pRight->pRight->pRight = NULL;

}

void PrintFrontToBottom(Tree *root)
{
	if(root == NULL)
	{
		return ;
	}
	queue<Tree *> queueTree;
	queueTree.push(root);
	while(queueTree.size())
	{
		Tree *temp = queueTree.front();
		queueTree.pop();
		cout<<temp->nValue<<" ";
		if(temp->pLeft)
		{
			queueTree.push(temp->pLeft);
		}
		if(temp->pRight)
		{
			queueTree.push(temp->pRight);
		}
	}
}
struct Queue
{
	Tree *tree;
	Queue *pNext;
};
void PushQueue(Queue **pHead,Queue **pEnd,Tree *tree)
{
	if(pHead == NULL || pEnd == NULL)
	{
		return ;
	}
	Queue *temp = new Queue;
	temp->pNext = NULL;
	temp->tree = tree;
	if((*pHead) == NULL)
	{
		(*pHead) = temp;
	}
	else
	{
		(*pEnd)->pNext = temp;
	}
	(*pEnd) = temp;
}
Tree * PopQueue(Queue **pHead, Queue **pEnd)
{
	if(pHead == NULL || pEnd == NULL || (*pHead) == NULL || (*pEnd) == NULL)
	{
		return NULL;
	}
	Queue *pDel = (*pHead);
	Tree *tree = pDel->tree;
	(*pHead) = (*pHead)->pNext;
	delete pDel;
	pDel = NULL;
	return tree;
}
void Print(Tree *root)
{
	Queue *pHead = NULL;
	Queue *pEnd = NULL;
	while(root)
	{
		cout<<root->nValue<<" ";
		if(root->pLeft)
		{
			PushQueue(&pHead,&pEnd,root->pLeft);
		}
		if(root->pRight)
		{
			PushQueue(&pHead,&pEnd,root->pRight);
		}
		root = PopQueue(&pHead,&pEnd);
	}
}

int main()
{
	Tree *root = NULL;
	CreateTree(&root);
	PrintFrontToBottom(root);
	cout<<endl;
	Print(root);
	system("pause");
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值