【二叉树】分层遍历二叉树和打印二叉树第k层节点

时间:2022-07-27 11:11:10

题目:《编程之美》P252 3.10

//分层遍历二叉树
void print_tree_by_level(const shared_ptr<treenode> root)
{
	if(root==nullptr)
		return;
	vector<shared_ptr<treenode>> nodes;
	nodes.push_back(root);
	int pcur=0,pend=nodes.size();
	while(pcur<pend)
	{		
		for(;pcur<pend;pcur++)
		{
			if(nodes[pcur]->left!=nullptr)
				nodes.push_back(nodes[pcur]->left);
			if(nodes[pcur]->right!=nullptr)
				nodes.push_back(nodes[pcur]->right);
		}
		pend=nodes.size();
	}

	for(int i=0;i<nodes.size();i++)
		cout<<nodes[i]->data<<"\t";
	cout<<endl;
}

//打印二叉树第k层的节点,k>=0
bool print_tree_level(const shared_ptr<treenode> root,const int k)
{
	if(root==nullptr || k<0)
		return false;
	if(k==0)
	{
		cout<<root->data<<"\t";
		return true;
	}
	bool b1=print_tree_level(root->left,k-1);
	bool b2=print_tree_level(root->right,k-1);
	return b1||b2;
}