bzoj 3124: [Sdoi2013]直径 （dfs）

3124: [Sdoi2013]直径

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 1032   Solved: 468
[ Submit][ Status][ Discuss]

Description

小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。

Source

[ Submit][ Status][ Discuss] 
HOME   Back

题解：dfs

两边dfs求出其中一条直径。

然后可以发现所有的直径都经过的边一定在找出的路径上，并且一定是路径上连续的一点。

然后从一端开始dfs，强制不走找出的直径上的边，如果从当前点开始的最长链等于改点到起始点的距离，那么这个点一定不是所有的直径都经过的边的端点，指针后移，然后判断当前点到另一端的距离是否等于最长链，如果等于就结束搜索，剩下那边的所有边一定在答案里，计算找到的点和指针之间有多少边即可。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#define N 400003
#define LL long long 
using namespace std;
int n,m,f[N],g[N],q[N],cnt,ansx;
int point[N],next[N],v[N],tot,mark[N],last[N];
LL dis[N],c[N],ans;
void add(int x,int y,int z)
{
	tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
	tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
	//cout<<x<<" "<<y<<endl;
}
void dfs(int x,int fa,LL dis)
{
	if (dis>ans)
	 {
	 	ans=dis; ansx=x;
	 }
	for (int i=point[x];i!=-1;i=next[i])
	 if (v[i]!=fa&&!mark[i])
	  {
	  	last[v[i]]=i;
	  	dfs(v[i],x,dis+c[i]);
	  }
}
void change(int x,int y)
{
	int now=y; dis[1]=0;
	while (now!=x)
	 {
	 	q[++cnt]=now; 
	 	mark[last[now]]=1;
	 	mark[last[now]^1]=1;
	 	dis[cnt+1]=dis[cnt]+c[last[now]];
	 	now=v[last[now]^1];
	 }
	q[++cnt]=x;
}
int main()
{
	freopen("a.in","r",stdin);
	scanf("%d",&n);
	tot=-1;
	memset(point,-1,sizeof(point));
	memset(next,-1,sizeof(next));
	for (int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
	}
	dfs(1,0,0); int head=ansx;
	ans=0;
	dfs(ansx,0,0); int tail=ansx;
	change(head,tail); LL ok=ans;
	int l=1;
	printf("%lld\n",ok);
	for (int i=1;i<=cnt;i++)
     {
     	ans=0;
     	dfs(q[i],0,0);
     	if (ans==dis[i])  l=i;
     	if (ans==ok-dis[i])  {
     		printf("%d\n",i-l);
     		return 0;
		 }
	 }
}