【题解】Luogu P3304 [SDOI2013] 直径 树的直径

暴力大法吼啊

先跑两遍$dfs$求树的直径，记录下直径上的点，暴力跑从每个点出发不经过直径的的树上最远距离点的路径（领会精神

最后都经过的路径就是答案

某位大佬说的

理论上复杂度$O(n*sum)$当树是一条链时会退化(?)成$O(n^2)$

然而某谷数据水，没判也能A

code

#include <bits/stdc++.h>
using namespace std;
namespace gengyf{
#define ll long long
#define int long long
const int maxn=2e5+10;
const int mod=1e7+7;
inline int read(){
    int f=1,x=0;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
    return f*x;
}
int n,m;
struct edge{
    int to,nxt,w;
}e[maxn*2];
int head[maxn],cnt,d[maxn],vis[maxn],fa[maxn];
int path[maxn],l,r,val[maxn],sum;
inline void add(int from,int to,int w){
    e[++cnt].to=to;e[cnt].w=w;e[cnt].nxt=head[from];head[from]=cnt;
}
void Dfs(int x,int len,int fa){
    l=max(l,len);
    for(int i=head[x];i;i=e[i].nxt){
        if(e[i].to==fa||vis[e[i].to])continue;
        vis[e[i].to]=1;Dfs(e[i].to,len+e[i].w,x);
    }
}
void dfs(int x){
    for(int i=head[x];i;i=e[i].nxt){
        if(e[i].to!=fa[x]){
            d[e[i].to]=d[x]+e[i].w;
            fa[e[i].to]=x;dfs(e[i].to);
        }
    }
}
void getd(){
    l=1;dfs(1);
    for(int i=2;i<=n;i++){
        if(d[i]>d[l])l=i;
    }
    d[l]=0;memset(fa,0,sizeof(fa));
    dfs(l);
    for(int i=2;i<=n;i++){
        if(d[i]>d[r])r=i;
    }
}
int main(){
    n=read();
    for(int i=1;i<n;i++){
        int u,v,w;
        u=read();v=read();w=read();
        add(u,v,w);add(v,u,w);
    }
    getd();printf("%lld\n",d[r]);
    memset(vis,0,sizeof(vis));
    for(int i=r;i;i=fa[i]){
        vis[i]=1;
    }
    for(int i=r;i;i=fa[i]){
        l=0;Dfs(i,0,0);val[i]=l;
    }
    int j=r;
    for(int i=fa[r];i;i=fa[i]){
        path[i]=j;j=i;
    }
    int i;
    for(i=j;i;i=path[i]){
        if(d[r]-d[i]==val[i])break;
    }
    for(;i;i=fa[i]){
        if(d[i]==val[i])break;
        sum++;
    }
    printf("%lld",sum);
    return 0;
}
}
signed main(){
  gengyf::main();
  return 0;
}