POJ1985+POJ2631 求一棵树的直径

这两个题目都是求解一棵树的直径，也就是书里面的最长的一段距离。

树的直径是指树的最长简单路。求法: 两遍BFS（dfs） :先任选一个起点BFS（dfs）找到最长路的终点，再从终点进行BFS（dfs），则第二次BFS（dfs）找到的最长路即为树的直径。

证明网上有很多，在此就不写了。

下面分别列出两种方法，一种是bfs，一种是dfs。

//POJ 1985

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;

const int N = 50000;
struct Node
{
int to,cap;
};
vector<Node> v
;
int vis
,dis
;
int ans;

int BFS(int x)
{
memset(dis,0,sizeof(dis));
memset(vis,0,sizeof(vis));
queue<int> q;
q.push(x);
vis[x]=1;
int point = 0;
while(!q.empty())
{
int f=q.front();
q.pop();
if(dis[f]>ans)
{
ans = dis[f];
point = f;
}
for(int i=0;i<v[f].size();i++)
{
Node tmp = v[f][i];
if(vis[tmp.to]==0)
{
vis[tmp.to]=1;
dis[tmp.to] = dis[f] + tmp.cap;
q.push(tmp.to);
}
}
}
return point;
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
for(int i=0;i<=n;i++)
v[i].clear();
for(int i=0;i<m;i++)
{
int x,y,z;
char c;
scanf("%d %d %d %c",&x,&y,&z,&c);
v[x].push_back((Node){y,z});
v[y].push_back((Node){x,z});
}
ans = 0;
int point = BFS(1);
ans = 0;
BFS(point);
printf("%d\n",ans);
}
return 0;
}

//POJ2631

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>
#include <queue>

using namespace std;

int const maxn = 100005;
struct edge
{
int v,next,val;
}e[maxn];
bool vis[maxn];
int n,k,ans,point;
int head[maxn];
//head[i]是以点i为起点的链表头部

void addedge(int a,int b,int w)
{//向图中加边的算法，注意加上的是有向边
e[k].v=a;
e[k].val=w;
e[k].next=head[b];
head[b]=k++;
}

void dfs(int u,int s)
{
vis[u]=1;
if(s>ans)
{
ans=s;
point=u;
}
for(int i=head[u];i>0;i=e[i].next)
{
int v=e[i].v;
if(vis[v]) continue;
dfs(v,s+e[i].val);
}
}

int main()
{
int a,b,w;
k = 1 ;
memset(head,0,sizeof(head));
while(cin>>a>>b>>w)
{
addedge(a,b,w);
addedge(b,a,w);
}

ans=0;
memset(vis,0,sizeof(vis));
dfs(1,0);

ans=0;
memset(vis,0,sizeof(vis));
dfs(point,0);

cout<<ans<<endl;

return 0;
}