【剑指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
****************************************************************/