poj1985 树的直径

题意求树的直径,直接贴代码
#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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值