桃花 (DFS)

桃花
时间限制: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;
}

猜你喜欢

转载自blog.csdn.net/qq_42819248/article/details/81949291
dfs