求二叉树的第K层结点数

//求二叉树的第K层结点数
#include
#include
using namespace std;
struct binaryTreeNode
{
	int value;
	binaryTreeNode *pLeft;
	binaryTreeNode *pRight;
};

int KthFloorNodeNum(binaryTreeNode *pRoot,int k)
{
	if(k<1 || pRoot==NULL)	return -1;
	if(1==k) return 1;
	int upFloorBegin=0,upFloorEnd=1;//End指向下一个结点
	vector pNodeVec;
	binaryTreeNode *pNode=NULL;
	pNodeVec.push_back(pRoot);
	for(int i=2;i<=k;i++)
	{
		for(int j=upFloorBegin;j!=upFloorEnd;j++)
		{
			pNode=pNodeVec[j];
			if(pNode->pLeft)
				pNodeVec.push_back(pNode->pLeft);
			if(pNode->pRight)
				pNodeVec.push_back(pNode->pRight);
		}
		upFloorBegin=upFloorEnd;
		upFloorEnd=pNodeVec.size();
	}
	return upFloorEnd-upFloorBegin;
}
int main()
{
	binaryTreeNode n1,n2,n3,n4,n5,n6,n7;
	n1.value=8,n2.value=6,n3.value=10,n4.value=5,n5.value=7,n6.value=9,n7.value=11;
	n1.pLeft=&n2,n1.pRight=&n3;
	n2.pLeft=&n4,n2.pRight=&n5;
	n3.pLeft=&n6,n3.pRight=&n7;
	n4.pLeft=NULL,n4.pRight=NULL;
	n5.pLeft=NULL,n5.pRight=NULL;
	n6.pLeft=NULL,n6.pRight=NULL;
	n7.pLeft=NULL,n7.pRight=NULL;
	binaryTreeNode *pRoot=&n1;
	cout<

你可能感兴趣的:(编程练习)