poj2631(树的直径)

解题思路:具体做法就是先随便选一个点,进行bfs或dfs,然后距离这个点最远的点,一定是树直径中的一个端点。然后再从这个端点bfs或dfs求得直径。

证明详见https://www.cnblogs.com/a-clown/p/6131346.html

#include<cstdio>
#include<cstring>
#include<vector>
#include<queue>
#define N 10009
using namespace std;
struct node
{
	int v,d;
	node(int v,int d):v(v),d(d){
	}
	node(){
	}
};
vector<node> s[N];
queue<node> q;
int vis[N],tp;
int bfs(int r)
{
	q.push(node(r,0));
	memset(vis,0,sizeof(vis));
	vis[r]=1;
	int k=-1;
	tp=0;
	while(!q.empty())
	{
		node t=q.front();
		q.pop();
		int u=t.v;
		if(t.d>tp)
		{
			tp=t.d;
			k=u;
		}
		for(int i=0;i<s[u].size();i++)
		{
			int v=s[u][i].v,d=s[u][i].d;
			if(!vis[v])
			{
				q.push(node(v,d+t.d));
				vis[v]=1;
			}
		}
	}
	return k;
}
int main()
{
	//freopen("t.txt","r",stdin);
	int a,b,c;
	while(scanf("%d%d%d",&a,&b,&c)!=EOF)
	{
		//printf("%d %d%d\n",a,b,c);
		s[a].push_back(node(b,c));
		s[b].push_back(node(a,c));
	}
	int r=bfs(1);
	bfs(r);
	printf("%d",tp);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_39861441/article/details/88380220
今日推荐