桃花
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
桃花一簇开无主,可爱深红映浅红。
——《题百叶桃花》
桃花长在桃树上,树的每个节点有一个桃花,调皮的HtBest想摘尽可能多的桃花。HtBest有一个魔法棒,摘到树上任意一条链上的所有桃花,由于HtBest法力有限,只能使用一次魔法棒,请求出Htbest最多可以摘到多少个桃花。
输入描述:
第一行有一个正整数n,表示桃树的节点个数。
接下来n-1行,第i行两个正整数ai,bi ,表示桃树上的节点ai,bi之间有一条边。
输出描述:
第一行一个整数,表示HtBest使用一次魔法棒最多可以摘到多少桃花。
示例1
输入
复制
3
1 2
2 3
输出
复制
3
示例2
输入
复制
3
1 2
1 3
输出
复制
3
示例3
输入
复制
4
1 2
2 3
3 4
输出
复制
4
备注:
对于100%的测试数据:
1 ≤ n ≤ 1000000
数据量较大,注意使用更快的输入输出方式。
解题思路:
需要求出最长的一条链,其实就是,首先我们将所有的点存入vecto数组中,即写出所有点的邻接矩阵,然后从任意一点开始搜索,求出经过该点的最长链,并记录该链上最后一点,从记录的那点开始重新搜索(因为有可能最长链并不经过我们初始找的点)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int maxn = 1e6 + 5;
vector<int>vec[maxn];//用于存放邻接矩阵
int vis[maxn];//用于记录是否已经搜索过该点
int ans , temp;
void dfs(int x, int h)
{
vis[x] = 1;//标记已经搜索过的点
if(h > ans)
{
ans = h;
temp = x;//记录最长链上的点
}
for(int i = 0; i < vec[x].size(); i++)
{
int t = vec[x][i];//t为接下来搜索的点
if(vis[t] == 0)
dfs(t, h+1);
}
return ;
}
int main()
{
int n, a, b, i;
scanf("%d", &n);
while(--n)
{
scanf("%d %d", &a, &b);
vec[a].push_back(b);//由于两点之间可以互通,所以需要将两点互相存入邻接点中
vec[b].push_back(a);
}
ans = 0;
memset(vis, 0, sizeof(vis));
dfs(1, 1);
memset(vis, 0, sizeof(vis));
dfs(temp, 1);
printf("%d\n", ans);
return 0;
}