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
#include
#include
#include
#include
#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<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



你可能感兴趣的:(dfs)