您的位置:首页 > 其它

poj1985 树的直径

2015-07-20 22:54 459 查看
题意求树的直径,直接贴代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
typedef struct
{
int to,val,next;   //定义边的终点以及权值和这条边的下一条边
}Edge;
Edge edge[100050];     //开50000会WA
int head[50000],e;
int dis[50000];        //记录第i个点的最长距离,若没有被访问过,则为-1

void addedge(int from,int to,int w){
edge[e].to = to;
edge[e].val = w;
edge[e].next = head[from];
head[from] = e++;    //保存下一条边的位置,e代表第e条边
}

int bfs(int v){
memset(dis,-1,sizeof(dis));
dis[v] = 0;        //初始点距离为0
int tem = v;
queue<int> q;
q.push(v);      //队列的入队
while(!q.empty())   //队列判断是否为空
{
int f = q.front();   //取队首元素
q.pop();    //队列弹出
for(int i = head[f]; i != -1; i = edge[i].next){     //对第f个点的邻接表遍历,i为第f个点导出的边
int t = edge[i].to;                           //第f个点能够到达的点
if(dis[t]==-1)    //如果没有被访问过
{
dis[t] = dis[f] + edge[i].val;    //更新距离
if(dis[t] > dis[tem])
tem = t;
q.push(t);
}
}
}
return tem;
}
int main(){
int m,n;
while(~scanf("%d%d",&n,&m)){
memset(head,-1,sizeof(head));       //邻接表存储图的表头,初始化为-1
e = 0;
int a,b,w;
char c;
for(int i = 0; i < m; i++)
{
scanf("%d%d%d %c",&a,&b,&w,&c);     //char型变量的输入,注意getchar
addedge(a,b,w);           //无向图加入两次
addedge(b,a,w);
}
int fa = bfs(1);             //第一次bfs求出一个最远的点
int ans = bfs(fa);           //第二次bfs从最远的点求另一个最远的点
printf("%d\n",dis[ans]);
}
}
晚上装了win10,调戏了一会cortana,感觉挺好的,等正式版上线ing
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: