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



posted @ 2014-12-24 21:35  huoyao  阅读(285)  评论(0编辑  收藏  举报