《剑指Offer》面试题：二叉树的深度

题目描述：

输入一棵二叉树，求该树的深度。从根结点到叶结点依次经过的结点（含根、叶结点）形成树的一条路径，最长路径的长度为树的深度。

思路

此题的递归思想比较简单，直接用递归将根节点的深度转换为求左右子孩子的深度的较大值+1，根节点的深度等于max（左孩子的深度，右孩子的深度）。

/*


输入：
第一行输入有n，n表示结点数，结点号从1到n。根结点为1。 n <= 10。

接下来有n行，每行有两个个整型a和b，表示第i个节点的左右孩子孩子。a为左孩子，b为右孩子。当a为-1时，没有左孩子。当b为-1时，没有右孩子。

输出：
输出一个整型，表示树的深度。

样例输入：
3
2 3
-1 -1
-1 -1
样例输出：
2
*/
/*

思路： 用递归，根节点的深度等于max（左孩子的深度，右孩子的深度）。 
*/

//此题用数组存储树节点

#include<stdio.h>
#include<stdlib.h>
struct TreeNode{
    int pLeftChild;
    int pRightChild;
};
//递归实现
int depthInBTree(TreeNode* pRoot,int index){
    if(pRoot==NULL||index==-1){//直到左右孩子时，返回0
        return 0;
    }
    int leftDepth=depthInBTree(pRoot,pRoot[index].pLeftChild);
    int rightDepth=depthInBTree(pRoot,pRoot[index].pRightChild);
    return (leftDepth>rightDepth?leftDepth:rightDepth)+1;
}
int main(void){
    int n;
    while(scanf("%d",&n)!=EOF&&n>0&&n<=10){
        TreeNode *pRoot=(TreeNode *)malloc(n*sizeof(TreeNode));
        if(pRoot==NULL){
            exit(EXIT_FAILURE);

        }
        for(int i=0;i<n;i++){
            int leftIndex;
            int rightIndex;
            scanf("%d%d",&leftIndex,&rightIndex);
            if(leftIndex!=-1){
                pRoot[i].pLeftChild=leftIndex-1;
            }
            else{
                pRoot[i].pLeftChild=-1;
            }
            if(rightIndex!=-1){
                pRoot[i].pRightChild=rightIndex-1;
            }
            else{
                pRoot[i].pRightChild=-1;
            }
        }

        int depth=depthInBTree(pRoot,0); 
        printf("%d\n",depth);

    }
    return 0;
} 

运行结果如下：