剑指Offer - 九度1350 - 二叉树的深度

剑指Offer - 九度1350 - 二叉树的深度
2013-11-23 00:54

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

输入：
第一行输入有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

题意分析：
　　题目要求求出二叉树的深度，也就是根节点到最远叶结点的长度。思路很清楚，DFS。
　　递归关系为：

　　　　max_depth(NULL) = 0;
　　　　max_depth(r) = max(max_depth(r->left), max_depth(r->right)) + 1;

　　实际上由于树的节点个数一开始就指定了，所以直接用数组存数据，用下标当指针就行了，coding更方便。题目中节点数n的数据范围居然只有10，我都怀疑应该是10000或者1000错写成10了。
　　但有一点要注意，根节点如果不指定为1的话，是需要我们自己去判断哪个是根节点的。对于任何树结构，都只有根节点的入度为零，因为没有节点指向根节点。找出根节点才能去递归。
　　每个节点遍历一次，时间复杂度O(n)。节点数据都要存起来，空间复杂度O(n)。

// 654348    zhuli19901106    1350    Accepted    点击此处查看所有case的执行结果    1020KB    982B    0MS
// 201311210410
#include <cstdio>
using namespace std;

int mymax(const int &a, const int &b)
{
return a > b ? a : b;
}

int max_depth(const int a[][2], int r)
{
if(a == NULL || r < 0){
return 0;
}

return mymax(max_depth(a, a[r][0]), max_depth(a, a[r][1])) + 1;
}

int main()
{
const int MAXN = 20;
int a[MAXN][2];
int in_degree[MAXN];
int i, n;

while(scanf("%d", &n) == 1){
if(n < 1){
printf("0\n");
continue;
}

for(i = 1; i <= n; ++i){
in_degree[i] = 0;
}

for(i = 1; i <= n; ++i){
scanf("%d%d", &a[i][0], &a[i][1]);
if(a[i][0] >= 1 && a[i][0] <= n){
++in_degree[a[i][0]];
}
if(a[i][1] >= 1 && a[i][1] <= n){
++in_degree[a[i][1]];
}
}

for(i = 1; i <= n; ++i){
if(in_degree[i] == 0){
break;
}
}
printf("%d\n", max_depth(a, i));
}

return 0;
}