【最长链】wikioi1814

最长链 

现给出一棵N个结点二叉树，问这棵二叉树中最长链的长度为多少，保证了1号结点为二叉树的根。

输入描述

输入的第1行为包含了一个正整数N，为这棵二叉树的结点数，结点标号由1至N。

接下来N行，这N行中的第i行包含两个正整数l[i], r[i]，表示了结点i的左儿子与右儿子编号。如果l[i]为0，表示结点i没有左儿子，同样地，如果r[i]为0则表示没有右儿子。

输出描述

输出包括1个正整数，为这棵二叉树的最长链长度。

样例输入

5

2 3

4 5

0 6

0 0

0 0

样例输出

4

做完之前那道题于是我找了一个最长链的题做

就是代码 依然很丑

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;

struct ty
{
    long t, next;
}edge[200010];
long head[100010];
bool v[100010];
long dist[100010];
long n, m, maxn, dian;
void insertedge(long x, long y, long k)
{
    edge[k].t = y;
    edge[k].next = head[x];
    head[x] = k;
}
void dfs(long x)
{
    long i = head[x];
    for (long i = head[x]; i != 0; i = edge[i].next)
        if (!v[edge[i].t])
        {
            v[edge[i].t] = true;
            dist[edge[i].t] = dist[x] + 1;
            dfs(edge[i].t);
            v[edge[i].t] = false;
        }
}
int main()
{
    freopen("1814.in", "r", stdin);
    scanf("%d", &n);
    m = 0;
    for (long i = 1; i <= n; i++)
    {
        long t1, t2;
        scanf("%d%d", &t1, &t2);
        if (t1 != 0)
        {
            insertedge(i, t1, ++m);
            insertedge(t1, i, ++m);
        }
        if (t2 != 0)
        {
            insertedge(i, t2, ++m);
            insertedge(t2, i, ++m);
        }
    }

    memset(dist, 0, sizeof(dist));
    memset(v, 0, sizeof(v));
    v[1] = true;
    dfs(1);
    maxn = 0, dian = 0;
    for (long i =1; i<=n ;i++)
    {
        if (dist[i] > maxn)
        {
            maxn = dist[i];
            dian = i;
        }
    }
    memset(dist, 0, sizeof(dist));
    memset(v, 0, sizeof(v));
    v[dian] = true;
    dfs(dian);
    maxn = 0, dian = 0;
    for (long i =1; i<=n ;i++)
    {
       // cout << dist[i]<<' ';
        if (dist[i] > maxn)
        {
            maxn = dist[i];
            dian = i;
        }
    }
    cout << maxn <<endl;
    return 0;
}