【剑指Offer面试编程题】 题目1350：二叉树的深度--九度OJ

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

输入：
第一行输入有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就能解决。难点在于树的构建和根结点的寻找。

AC code：

#include <cstdio>
#include <vector>
using namespace std;

struct tr
{
int lc,rc;
};

void dfs(const vector<tr> &tre,const int& id,int &maxcnt,int &cnt)
{
++cnt;
if(tre[id].lc==-1 && tre[id].rc==-1)
{
if(cnt>maxcnt) maxcnt=cnt;
return ;
}
if(tre[id].lc!=-1) {dfs(tre,tre[id].lc,maxcnt,cnt);--cnt;}
if(tre[id].rc!=-1) {dfs(tre,tre[id].rc,maxcnt,cnt);--cnt;}
}

int main()
{
int n,a,b;
scanf("%d",&n);
vector<tr> tre(n+1);
vector<int> vecidx(n+1,0);
tr rt;
for(int i=1;i<=n;++i)
{
scanf("%d%d",&rt.lc,&rt.rc);
tre[i]=rt;
if(rt.lc!=-1)vecidx[rt.lc]=1;
if(rt.rc!=-1)vecidx[rt.rc]=1;
}
int i=1;
for(;i<=n;++i)
if(vecidx[i]==0) break;
int maxcnt=0,cnt=0;
if(n!=0) dfs(tre,i,maxcnt,cnt);
printf("%d\n",maxcnt);
return 0;
}
/**************************************************************
Problem: 1350
User: huo_yao
Language: C++
Result: Accepted
Time:0 ms
Memory:1052 kb
****************************************************************/