学习笔记28——按之字形打印二叉树

题目:请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
思路:按之字形打印二叉树需要两个栈,在打印某一层的节点时,把下一层的子节点保存到相应的栈里。
1.如果当前打印的是奇数层,则先保存左子节点再保存右子节点;如果当前打印的是偶数层,则先保存右子节点再保存左子节点
2.在打印一个栈里的节点时,它的子节点保存到另一个栈里
3.当一层所有节点都打印完毕时,交换这两个栈并继续打印下一层
核心代码如下:

struct BinaryTreeNode{
	int m_nValue;
	BinaryTreeNode* m_pLeft;
	BinaryTreeNode* m_pRight;
};

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

	stack<BinaryTreeNode*> levels[2]; //创建一个栈数组,包含 levels[0]和 levels[1]
	int current = 0;
	int next = 1;

	levels[current].push(pRoot);
	while(!levels[0].empty() || !levels[1].empty()){
		BinaryTreeNode* pNode = levels[current].top();
		levels[current].pop();
		cout << pNode->m_nValue;

		if(current == 0){
			//先保存左节点,后保存右节点
			if(pNode->m_pLeft != nullptr)
				levels[next].push(pNode->m_pLeft);
			if(pNode->m_pRight != nullptr)
				levels[next].push(pNode->m_pRight);
		}
		else{
			//先保存右节点,后保存左节点
			if(pNode->m_pRight != nullptr)
				levels[next].push(pNode->m_pRight);
			if(pNode->m_pLeft != nullptr)
				levels[next].push(pNode->m_pLeft);
		}

		if(levels[current].empty()){
			cout << endl;
			current = 1 - current; //交换两个栈
			next = 1 - next;
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值