直径 HYSBZ - 3124（树的直径）

小Q最近学习了一些图论知识。根据课本，有如下定义。树：无回路且连通的无向图，每条边都有正整数的权值来表示其长度。如果一棵树有N个节点，可以证明其有且仅有N-1 条边。 路径：一棵树上，任意两个节点之间最多有一条简单路径。我们用 dis(a,b)
表示点a和点b的路径上各边长度之和。称dis(a,b)为a、b两个节点间的距离。
直径：一棵树上，最长的路径为树的直径。树的直径可能不是唯一的。
现在小Q想知道，对于给定的一棵树，其直径的长度是多少，以及有多少条边满足所有的直径都经过该边。

Input
第一行包含一个整数N，表示节点数。
接下来N-1行，每行三个整数a, b, c ，表示点 a和点b之间有一条长度为c
的无向边。

Output

共两行。第一行一个整数，表示直径的长度。第二行一个整数，表示被所有
直径经过的边的数量。

Sample Input

6
3 1 1000
1 4 10
4 2 100
4 5 50
4 6 100
Sample Output
1110
2

【样例说明】
直径共有两条，3 到2的路径和3到6的路径。这两条直径都经过边(3, 1)和边(1, 4)。
Hint
对于100%的测试数据：2≤N≤200000，所有点的编号都在1..N的范围内，

边的权值≤10^9。

此题莫名re，跪求大佬指出错误。
代码：

import java.util.Scanner;

public class Main
{   
    static int maxx=200005;
    static int head[]=new int[maxx];
    static int to[]=new int[maxx<<1];
    static int w[]=new int[maxx<<1];
    static int next[]=new int[maxx<<1];
    static int cnt;
    static void addEdge(int x,int y,int _w)
    {
        to[++cnt]=y;w[cnt]=_w;next[cnt]=head[x];head[x]=cnt;
        to[++cnt]=x;w[cnt]=_w;next[cnt]=head[y];head[y]=cnt;
    }
    static int n;
    static long d[]=new long[maxx];
    static int fa[]=new int[maxx];
    static void dfs(int x,int last)
    {
        fa[x]=last;
        for(int i=head[x];i>0;i=next[i])
        {
            int y=to[i];
            if(y==last)
                continue;
            d[y]=d[x]+w[i];
            dfs(y,x);
        }
    }
    static boolean mark[]=new boolean [maxx];
    static long maxD;

    static void find(int x,long D)
    {
        for(int i=head[x];i>0;i=next[i])
        {
            int y=to[i];
            if(!mark[y])
            {
                mark[y]=true;
                maxD=Math.max(D+w[i],maxD);
                find(y,D+w[i]);
            }
        }
    }
    public static void main(String[] args)
    {
        Scanner sc=new Scanner(System.in);
        n=sc.nextInt();
        for(int i=1;i<n;i++)
            addEdge(sc.nextInt(),sc.nextInt(),sc.nextInt());//建边
        dfs(1,0);
        int s=1;
        for(int i=2;i<=n;i++)
            if(d[s]<d[i])
                s=i;
        d[s]=0;
        dfs(s,0);//两次dfs找直径
        int e=1;
        for(int i=2;i<=n;i++)
            if(d[e]<d[i])
                e=i;
        System.out.println(d[e]);
        int temp=e;
        while(temp>0)
        {
            mark[temp]=true;
            temp=fa[temp];
        }//在直径上点给标记上
        int L=e,R=s;
        boolean sign=true;
        for(int i=fa[e];i!=s;i=fa[i])
        {
            long rD=d[i];
            long lD=d[e]-d[i];
            maxD=0;
            find(i,0);
            if(maxD==rD&&sign)//对于从右边的点，我们只要判断一次就好了，不然可能会是答案边的更大。
            {
                R=i;

                sign=false;
            }
            if(maxD==lD)L=i;//对于从左边开始的点，我不断找以这个点为根且点不在直径上的的最远距离，判断是否相等
        }
        for(int i=1;i<=n;i++)
            System.out.println(mark[i]);
        int ans=0;
        for(int i=L;i!=R;i=fa[i])
            ans++;
        System.out.println(ans);
    }
}