求二叉樹第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