C++-蓝桥杯-大臣的旅费[dfs][树的直径]

最坑的一点是没有数据范围，但是就蓝桥杯的德性，int就够了然后数组尽可能开大一点

O(n)算法要自信，嗯嗯嗯！

一次AC，没什么好说的，上代码：

#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1e6+10,M=2e6+10,INF=1e9;
struct edge{
    int v,w,next;
    edge(int v=0,int w=0,int next=0):v(v),w(w),next(next){}
};
edge E[M];
int head[N],cnt,n,dis[N];
void dfs(int u,int fa){
    for(int i=head[u];i;i=E[i].next)
        if(E[i].v!=fa){
            dis[E[i].v]=min(dis[E[i].v],dis[u]+E[i].w);
            dfs(E[i].v,u);
        }
}
int search(int root,int type){
    for(int i=1;i<=n;i++)dis[i]=INF;
    dis[root]=0;
    dfs(root,root);
    int ans=0,res;
    for(int i=1;i<=n;i++)if(dis[i]>ans)ans=dis[i],res=i;
    return type?res:ans;
}
int calc(int dis){
    int ans=dis*10,a=dis;
    if(a%2)a=(a+1)/2,ans+=a*dis;
    else a=a/2,ans+=a*(1+dis);
    return ans;
}
int main(){
    int u,v,w;
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&u,&v,&w);
        E[++cnt]=edge(v,w,head[u]),head[u]=cnt;
        E[++cnt]=edge(u,w,head[v]),head[v]=cnt;
    }
    int rt=search(1,1),dis=search(rt,0);
    printf("%d
",calc(dis));
    return 0;
}