《剑指Offer》学习笔记--面试题61:按之字形顺序打印二叉树

题目:请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,即第一行按照从左到右的顺序打印,第二层按照从右到左顺序打印,第三行再按照从左到右的顺序打印,其他以此类推。

按之字形顺序打印二叉树需要两个栈。我们在打印某一行结点时,把下一层的子结点保存到相应的栈里。如果当前打印的是奇数层,则先保存左子结点再保存右子结点到一个栈里;如果当前打印的是偶数层,则先保存右子结点再保存左子结点到第二个栈里。

void Print(BinaryTreeNode* pRoot)
{
	if(pRoot == NULL)
		return;

	std::stack<BinaryTreeNode*> levels[2];
	int current = 0;
	int next = 1;

	levels[current].push(pRoot);
	while(!levels[0].empty() || !levels[1].empty()){
		BianryTreeNode* pNode = levels[current].top();
		levels[current].pop();

		printf("%d", pNode->m_nValue);

		if(current == 0){
			if(pNode->m_pLeft != NULL)
				levels[next].push(pNode->m_pLeft);
			if(pNode->m_pRight != NULL)
				levels[next].push(pNode->m_pRight);
		}
		else{
			if(pNode->m_pRight != NULL)
				levels[next].push(pNode->m_pRight);
			if(pNode->m_pLeft != NULL)
				levels[next].push(pNode->m_pLeft);
		}

		if(levels[current].empty()){
			printf("\n");
			current = 1 - current;
			next = 1 - next;
		}
	}
}
上述代码定义了两个栈levels[0]和levels[1]。当打印一个栈里的结点时,它的子结点保存到另一个栈里。当一层所有结点都打印完了时,交换这两个站并继续打印下一层。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值