二叉树层序、层序换行、层序Z型打印

二叉树层序、层序换行、层序Z型打印

<span style="font-size:14px;">//节点数据结构
class TNode
{
public:
	TNode() { lchild = rchild = parent = nullptr; };
	T data;
	TNode<T>* lchild, *rchild, *parent;
};</span>

一：层序打印

利用队列的先进先出特效，教科书般的答案。

<span style="font-size:14px;">template<class T>
void TriTree<T>::LevalOrder()
{
	printf("层序遍历：");
	if (nullptr == root)
	{
		return;
	}
	TNode<T>* pNode = root;
	queue<TNode<T>*> QNode;
	QNode.push(pNode);
	while (!QNode.empty())
	{
		pNode = QNode.front();
		cout << pNode->data << " ";
		if (nullptr != pNode->parent)
		{
			cout << " parent:" << pNode->parent->data << endl;
		}
		QNode.pop();
		if (nullptr != pNode->lchild)
		{
			QNode.push(pNode->lchild);
		}
		if (nullptr != pNode->rchild)
		{
			QNode.push(pNode->rchild);
		}
	}
	puts("");
}</span>

二：层序换行打印。

只需要增加一个变量标记一行的结束就行，我们在每层遍历完后加入nullptr.当nullptr出队时，说明一行结束，输出换行。但是关键就在于加入nullptr时上层的所有节点必须已经入队。

<span style="font-size:14px;">template<class T>
void TriTree<T>::LevalChangeLineOrder(TNode<T>* bt)
{
	queue<TNode<T>*> QNode;
	TNode<T>* pTemp = nullptr;
	QNode.push(bt);
	QNode.push(nullptr);
	while (!QNode.empty())
	{
		pTemp = QNode.front();
		QNode.pop();
		if (nullptr != pTemp)
		{
			Visited(pTemp);
			if (nullptr != pTemp->lchild)
			{
				QNode.push(pTemp->lchild);
			}
			if (nullptr != pTemp->rchild)
			{
				QNode.push(pTemp->rchild);
			}
		}
		else if (!QNode.empty())
		{
			QNode.push(nullptr);
			puts("");
		}
	}
}</span>

三：层序Z型打印，就是上一行从左到右，下一行从右到左。

我看了别人上网博客多数是使用双端队列，但个人感觉不容易懂，而且双端队列在STL中是个效率很低的容器。具体请看STL源码。

1：使用vector和层序遍历一样，只不过将偶数行reserve一下，简单粗暴，但是很麻烦。

2：使用双栈，原理很简单，画图便可明白。

<span style="font-size:14px;">void TriTree<T>::LevalOrderZig(TNode<T>* pRoot)
{	//Zig型打印
	puts("ZigZag 打印");
	if (nullptr == pRoot)
	{
		return;
	}
	stack<TNode<T>*> SNodeLeft;
	stack<TNode<T>*> SNodeRight;
	TNode<T>* pTemp = nullptr;
	SNodeLeft.push(pRoot);
	while (!SNodeLeft.empty() || !SNodeRight.empty())
	{
		while (!SNodeLeft.empty())
		{
			pTemp = SNodeLeft.top();
			//将Left栈的元素从左孩子 到右孩子压栈
			SNodeLeft.pop();
			Visited(pTemp);
			if (nullptr != pTemp->lchild)
			{
				SNodeRight.push(pTemp->lchild);
			}
			if (nullptr != pTemp->rchild)
			{
				SNodeRight.push(pTemp->rchild);
			}
		}
		puts("");
		while (!SNodeRight.empty())
		{
			pTemp = SNodeRight.top();
			//将Left栈的元素从有孩子 到左孩子压栈
			SNodeRight.pop();
			Visited(pTemp);
			if (nullptr != pTemp->lchild)
			{
				SNodeLeft.push(pTemp->lchild);
			}
			if (nullptr != pTemp->rchild)
			{
				SNodeLeft.push(pTemp->rchild);
			}
		}
		puts("");
	}
}</span>