求二叉树第n层节点数

在知乎看到今日头条的一个面试题“求二叉树第n层节点数”：https://zhuanlan.zhihu.com/p/25671699，想到了这样一个解法，欢迎大家交流

我的解法采用递归的思想，从0层开始，逐层往下递归。然后达到递归终止条件时(cur == goal - 1)，就会把n-1层的所有儿子数都统计上来，代码如下：

int CountChildNum(Tree *t, int n)
{
    if(NULL == t)
        Error("fatal error");
    if(n == 0)
        return 1;
    return CountChild(t, 0, goal);
}

//cur:递归到的当前层， goal：目标层
int CountChild(Tree *t, int cur, int goal)
{
    if(NULL == t || cur >= goal)
        return 0;
    
    int c = 0;
    if(cur == goal - 1) //此时t的儿子即为想要的层数
    {
        if(t -> left != NULL)
            ++c;
        if(t -> right != NULL)
            ++c;
    }
    else
    {
        c += CountChild(t -> left, cur + 1, goal);
        c += CountChild(t -> right, cur + 1, goal);
    }
    return c;
}

这里有更好的解法：

http://www.cnblogs.com/hapjin/p/5505988.html

轻松搞定面试中的二叉树题目：

http://blog.csdn.net/luckyxiaoqiang/article/details/7518888