【最长链】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;
}