树的深度、宽度计算

#include "OJ.h"
#include <string.h>
#include <stdio.h>

/*****************计算树的深度*******************/
int TreeDepth(BiNode* pTop)
{
    if (pTop == NULL)
    {
        return 0;
    }
    else
    {
        int left = TreeDepth(pTop->left);
        int right = TreeDepth(pTop->right);
        return (left>right? left+1:right+1);
    }
}

void bt_level_num(BiNode* tree, int a[], int h) {
    if (tree == NULL)
    {
        h = 0;
    }
    else
    {
        a[h] += 1;
        bt_level_num(tree->left, a, h + 1);
        bt_level_num(tree->right, a, h + 1);
    }
}
/****************计算树的宽度**********************/
int bt_width(BiNode* tree) {
    int a[N]={0}, h = 0, max=0, i;

    bt_level_num(tree, a, h);

    for (i = 0; i < N; i++)
    {
        if (a[i] > max)
        {
            max = a[i];
        }
    }
    return max;
}


int GetBiNodeInfo(BiNode &head, unsigned int *pulWidth, unsigned int *pulHeight)
{
    /*在这里实现功能*/
    BiNode* pTop = &head;
    if (pTop == NULL)
    {
        *pulWidth = 0;
        *pulHeight = 0;
        return -1;
    }

    unsigned int depth = TreeDepth(pTop);
    *pulHeight = depth;

    unsigned int wd = bt_width(pTop);
    *pulWidth = wd;
    return 0;
}