树的直径

##树的直径

###定义：

树上距离最远的点（树的最长路）。

算法

思想

在树上任选一点u，求距离点u最远的点v，再求距离点v最远的点s，点v到点s的距离即为树的直径。

过程

两遍BFS :先任选一个起点BFS找到最长路的终点，再从终点进行BFS，则第二次BFS找到的最长路即为树的直径；

原理： 设起点为u,第一次BFS找到的终点v一定是树的直径的一个端点

证明

1. 如果u 是直径上的点，则v显然是直径的终点(因为如果v不是的话，则必定存在另一个点w使得u到w的距离更长，则 于BFS找到了v矛盾)
2. 如果u不是直径上的点，则u到v必然于树的直径相交(反证),那么交点到v 必然就是直径的后半段了所以v一定是直径的一个端点，所以从v进行BFS得到的一定是直径长度

题目

poj2631 http://poj.org/problem?id=2631 （裸题）

#include<bits/stdc++.h>
using namespace std;
int head[100010],cnt,s,ans;
int d[100010],v[100010];
struct node
{
	int to,next,w;
} a[100010];
void add(int u,int vv,int w)
{
	cnt++;
	a[cnt].to=vv;
	a[cnt].w=w;
	a[cnt].next=head[u];
	head[u]=cnt;
}
void ddd(int x)
{
	int u,vv,i;
	memset(v,0,sizeof(v));
	queue<int>q;
	q.push(x);
	d[x]=0;
	v[x]=1;
	ans=0;
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		for(i=head[u]; i!=-1; i=a[i].next)
		{
			vv=a[i].to;
			if(v[vv]==0)
			{
				v[vv]=1;
				d[vv]=d[u]+a[i].w;
				if(ans<d[vv])
				{
					ans=d[vv];//这个是最长距离
					s=vv;//距离当前点最远的点的编号
				}
				q.push(vv);
			}
		}
	}
}
int main()
{
	freopen("a.txt","r",stdin);
	int t,tt,n,x,y,z,i;
	scanf("%d",&tt);
	for(t=1; t<=tt; t++)
	{
		scanf("%d",&n);
		cnt=0;
		memset(head,-1,sizeof(head));
		for(i=1; i<n; i++)
		{
			scanf("%d%d%d",&x,&y,&z);
			add(x,y,z);
			add(y,x,z);
		}
		ddd(0);
		ddd(s);
		printf("Case %d: %d\n",t,ans);
	}
	return 0;
}

树的直径的个数

题解

1. 求最长边的时候可以统计一下最长边的个数。
2. 找到必经边的左右边界，然后两个边界点向两边走，各自算一下有多少条最长路，乘一下。

题目

HDU 3534 http://acm.hdu.edu.cn/showproblem.php?pid=3534

第一种方法

#include<stdio.h>
#include<string.h>
const int N=500000;
const int inf=0x3fffffff;
int head[N],num,Len[N],ans,node[N],k;
struct edge
{
	int ed,w,next;
}e[N*2];
void addedge(int x,int y,int w)
{
	e[num].ed=y;e[num].w=w;e[num].next=head[x];head[x]=num++;
	e[num].ed=x;e[num].w=w;e[num].next=head[y];head[y]=num++;
}
void dfs(int u,int fa)
{
	int i,v,temp;
	Len[u]=0;//最长边
	node[u]=1;//最长边的个数
	for(i=head[u];i!=-1;i=e[i].next)
	{
		v=e[i].ed;
		if(v==fa)continue;
		dfs(v,u);
		temp=Len[v]+e[i].w;
		if(temp+Len[u]>ans)//最长边经过v
		{
			k=node[v]*node[u];
			ans=temp+Len[u];
		}
		else if(temp+Len[u]==ans)
			k+=node[v]*node[u];
		if(Len[u]<temp)//更新最长边
		{
			Len[u]=temp;
			node[u]=node[v];
		}
		else if(Len[u]==temp)//更新最长边的个数
			node[u]+=node[v];
	}
}
int main()
{
	int n,i,x,y,w;
	while(scanf("%d",&n)!=-1)
	{
		memset(head,-1,sizeof(head));
		num=0;
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&x,&y,&w);
			addedge(x,y,w);
		}
		ans=-inf;k=0;
		dfs(1,0);
		printf("%d %d\n",ans,k);
	}
	return 0;
}

第二种方法

#include<bits/stdc++.h>
using namespace std;
int head[400010],cnt,s,ans,num,mmax,s1;
int d[400010],v[400010],p[400010];
struct node
{
	int to,next,w;
} a[400010];
void add(int u,int vv,int w)
{
	cnt++;
	a[cnt].to=vv;
	a[cnt].w=w;
	a[cnt].next=head[u];
	head[u]=cnt;
}
void ddd(int x)
{
	int u,vv,i;
	memset(v,0,sizeof(v));
	queue<int>q;
	q.push(x);
	d[x]=0;
	v[x]=1;
	ans=0;
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		for(i=head[u]; i!=-1; i=a[i].next)
		{
			vv=a[i].to;
			if(v[vv]==0)
			{
				v[vv]=1;
				d[vv]=d[u]+a[i].w;
				if(ans<d[vv])
				{
					ans=d[vv];//这个是最长距离
					s=vv;//距离当前点最远的点的编号
					s1=1;
				}
				else if(ans==d[vv]) s1++;
				q.push(vv);
			}
		}
	}
}
void dfs(int x,int y)
{
	p[++num]=x;
	v[x]=1;
	for(int i=head[x]; i!=-1; i=a[i].next)
	{
		if(d[a[i].to]==y-a[i].w) dfs(a[i].to,y-a[i].w);
	}
}
void dfs1(int x,int fa,int z)
{
	mmax=max(mmax,z);
	for(int i=head[x]; i!=-1; i=a[i].next)
	{
		if(a[i].to==fa||v[a[i].to]) continue;
		dfs1(a[i].to,x,z+a[i].w);
	}

}
int main()
{
	freopen("a.txt","r",stdin);
	int n,x,y,z,i;
	scanf("%d",&n);
	cnt=0;
	num=0;
	memset(head,-1,sizeof(head));
	for(i=1; i<n; i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	ddd(1);
	ddd(s);
	printf("%d\n",ans);
	memset(v,0,sizeof(v));
	dfs(s,ans);
	int l=1,r=num;
	for(i=2; i<num; i++)
	{
		mmax=0;
		dfs1(p[i],0,0);
		if(mmax==ans-d[p[i]]) l=i;
		if(mmax==d[p[i]])
		{
			r=i;
			break;
		}
	}
	ans=0;
	s1=0;
	memset(v,0,sizeof(v));
	for(i=l; i<=r; i++) v[p[i]]=1;
	ddd(p[l]);
	int s2=s1;
	ans=0;
	s1=0;
	memset(v,0,sizeof(v));
	for(i=l; i<=r; i++) v[p[i]]=1;
	ddd(p[r]);
	printf("%d\n",s1*s2);
	return 0;
}