分层遍历二叉树

用map存储节点 <key ,value> key表示节点编号, value 是儿子,用pair实现。


遍历采用 bfs 、 优化在于 level 部分

#include<iostream>
#include<sstream>
#include<string>
#include<map>
#include<queue>
using namespace std;

typedef pair<int,int> Son;
typedef map<int,Son> Tree;

void bfs(Tree t,int root)
{
	queue<int> Q;
	Q.push(root);
	int level = 1;

	while( !Q.empty() ) {
		cout<<"level:"<<level<<endl;
		level++;
		
		int numNode = Q.size();
		while( numNode-- ){
			int node = Q.front(); 	Q.pop();
			cout<<node<<"  ";

			Tree::iterator pos = t.find(node);
			if (pos != t.end()){
				Son child = pos->second;
				if( child.first >0) Q.push( child.first );
				if( child.second>0) Q.push( child.second);
			}
		}

		cout<<endl;
	}
}

int main()
{
	Tree t;
	int root;
	cout<<"root:";
	cin>>root;	cin.get();

	string line;
	while( getline(cin,line) ){
		if (line == "0") break;
		stringstream str(line);

		int father;
		str>>father; 
		cout<<father <<endl;
		
		int lchild, rchild;
		lchild = rchild = -1;
		str >> lchild >> rchild;
		cout<<lchild <<","<<rchild <<endl; 

		if( lchild!=-1 || rchild!=-1)
			t[father] = Son(lchild,rchild); 
	}

	bfs(t,root);

	system("pause");
	return 0; 
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值