【演算法】輸入一顆二元樹,從上往下按層列印樹的每個結點,同一層中按照從左往右的順序列印

pengfoo發表於2012-10-01

題目:輸入一顆二元樹,從上往下按層列印樹的每個結點,同一層中按照從左往右的順序列印。

例如輸入

      8
    /  \
   6    10
  /\     /\
5  7   9  11

輸出8   6   10   5   7   9   11。

思路是遍歷一個結點時,首先訪問它,然後將它的左右子樹放入佇列中。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <queue>
using namespace std;

typedef struct node
{
	int key;
	struct node *pleft;
	struct node *pright; 
}Node;

int CreateTreeByInsertData(Node **p,int k)//理解為什麼用二級指標
{
	if(*p==NULL)      
	{
		*p=(Node *)malloc(sizeof(Node));	
		(*p)->key=k; 
		(*p)->pleft=(*p)->pright=NULL;     
		return 1;
	}
	else if(k == (*p)->key)          
		return 0;
	else if(k < (*p)->key)           
		return CreateTreeByInsertData(&(*p)->pleft,k); 
	else
		return CreateTreeByInsertData(&(*p)->pright,k);  

}

void visitByLevel(Node *p)
{
	queue<Node*> myQueue;
	if(p == NULL)
		return;
	myQueue.push(p);
	while(!myQueue.empty())
	{
		Node *now = myQueue.front();
		myQueue.pop();
		printf("%d ",now->key);
		if(now->pleft) myQueue.push(now->pleft);
		if(now->pright) myQueue.push(now->pright);
	}
	printf("\n");
}


void ClearTree(Node** tree)//刪除樹的操作,在本題中不一定用的到
{
	if(*tree==NULL)return;
	ClearTree(&(*tree)->pleft);
	ClearTree(&(*tree)->pright);
	free(*tree);
	*tree=NULL;
}


int main()
{
	int i;
	Node *proot = NULL;
	
	int data[] = {8,6,10,5,7,9,11};

	//依次插入一些資料,建立一個二叉排序樹
	for(i=0; i<sizeof(data)/sizeof(int); i++)
		CreateTreeByInsertData(&proot, data[i]);
	visitByLevel(proot);

	ClearTree(&proot);

	return 0; 
}


 

相關文章