分层遍历二叉树

#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <stdio.h>
#include <time.h>

using namespace std;

struct Node
{
	int data;      // 节点中的数据
	Node *m_pLeft;
	Node *m_pRight;
};

Node* CreatNode(int value)
{
	Node* pNode = new Node;
	pNode->data = value;
	return pNode;
}

void ConnectNodes(Node* pParent, Node* pLeft, Node* pRight)
{
	if (pParent != nullptr)
	{
		pParent->m_pLeft = pLeft;
		pParent->m_pRight = pRight;
	}
}


void PrintNodeByLevel(Node* root)
{
	if (root == nullptr)
	{
		return;
	}

	vector<Node*> NodeVec;
	NodeVec.push_back(root);

	int cur = 0;
	int last = 1;

	while (cur < NodeVec.size())//更新cur
	{
		last = NodeVec.size();//更新last
		while (cur < last)//当前层的遍历输出
		{
			cout << NodeVec[cur]->data << " ";
			if (NodeVec[cur]->m_pLeft)
			{
				NodeVec.push_back(NodeVec[cur]->m_pLeft);
			}
			if (NodeVec[cur]->m_pRight)
			{
				NodeVec.push_back(NodeVec[cur]->m_pRight);
			}
			cur++;
		}
		cout << endl;
	}
}

int main()
{
	//            1
	//          /   \
	//        2      3
	//       /  \      \
	//      4   5       6
	//         /  \
	//        7    8
	Node *pNode1 = CreatNode(1);
	Node *pNode2 = CreatNode(2);
	Node *pNode3 = CreatNode(3);
	Node *pNode4 = CreatNode(4);
	Node *pNode5 = CreatNode(5);
	Node *pNode6 = CreatNode(6);
	Node *pNode7 = CreatNode(7);
	Node *pNode8 = CreatNode(8);

	ConnectNodes(pNode1, pNode2, pNode3);
	ConnectNodes(pNode2, pNode4, pNode5);
	ConnectNodes(pNode5, pNode7, pNode8);
	ConnectNodes(pNode3, nullptr, pNode6);
	ConnectNodes(pNode4, nullptr, nullptr);
	ConnectNodes(pNode6, nullptr, nullptr);
	ConnectNodes(pNode7, nullptr, nullptr);
	ConnectNodes(pNode8, nullptr, nullptr);

	PrintNodeByLevel(pNode1);

	return 0;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Digital2Slave

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值