统计二叉树每层节点个数并打印每层节点

先看百度一道面试题：

有一颗二叉树，定义树的高度为从根到叶子节点的最长距离，树的宽度为每层节点的最大值，树的面积定义为高度和宽度的乘积。写一个函数计算一个二叉树的面积。(15分)

解决这道题参考资料剑指offer和编程之美，分别求求二叉树高度和宽度。

都是采用递归：

求高度参考：http://blog.csdn.net/buyingfei8888/article/details/38345591

求每层节点个数需要先求出高度

代码：

#include <iostream>
using namespace std;

typedef struct node{
int data;
struct node *lchild;
struct node *rchild;
}Node ,*pNode;

void createTree(pNode & root){
int temp;
scanf("%d",&temp);
if(0 != temp){
root=(pNode) malloc (sizeof(Node));
root->data = temp;
createTree(root->lchild);
createTree(root->rchild);
}else{
root=NULL;
}

}

void printAll(const pNode root){
pNode p = root;
if(p){
cout<< p->data<< " ";
printAll(p->lchild);
printAll(p->rchild);
}

}
//打印一层节点
int printNodeAtLevel(pNode root,int level,int &count){
if(!root || level <0)
return 0;
if(0 == level){
cout<< root->data<< " ";
count++;
return 1;
}
return printNodeAtLevel(root->lchild,level-1,count) + printNodeAtLevel(root->rchild,level-1,count);
}

//类似后序遍历，效率比较高，不会重复访问节点
void btDepth(const pNode root,int &depth){
if(NULL == root){
depth=0;
return ;
}
int left ;
btDepth(root->lchild,left);

int right ;
btDepth(root->rchild,right);
depth = left > right ? left +1 : right +1;
}
int main(){
pNode root=NULL;
createTree(root);
cout<<endl;
printAll(root);
cout<<endl;
int depth=0;
btDepth(root,depth);
cout<<endl<<depth;
cout<<endl;

//循环遍历 ，打印各层几点
for(int i=0;i<depth;i++){
int count=0;
printNodeAtLevel(root,i,count);
cout << " 节点个数为："<< count<<endl;
}
return 0;

}

运行结果：