ZJUT12-DAY2 树的直径

今天又是复习（预习）的一天呢，码一个树的直径，虽然题目没有边权，不过这个代码只要把权值从1改为原本的权值就可以了。另外，它也可以用于判断有多少条直径，只要记录一下第一次的最大点有几个，第二次有几个，重复的有几个（其实一旦重复，必然是第一次的个数减一）这非常的方便。题目链接SP1437 PT07Z - Longest path in a tree

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 5;
struct edge
{
    int to,cost;
    edge(int a,int b):to(a),cost(b){};
};
vector<edge> p[maxn];
bool used[maxn];
int ed;
int mx;
int dfs1(int now,int len)
{
    bool flag=false;
    used[now]=1;
    for(int i=0;i<p[now].size();i++)
    {
        if(!used[p[now][i].to])
        {
            flag=true;
            dfs1(p[now][i].to,len+p[now][i].cost);
        }
    }
    if(!flag)
    {
        if(mx<len)
        {
            ed=now,mx=len;
        }
    }
    return 0;
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<n;i++)
    {
        int a,b,c;
        cin>>a>>b;
        p[a].push_back(edge(b,1));
        p[b].push_back(edge(a,1));
    }
    dfs1(1,0);
    memset(used,0,sizeof(used));
    dfs1(ed,0);
    cout<<mx<<endl;


}