小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);
}
}