HDU4514

树的直径

并查集判环,SPFA判环一直T。。。。(写的太挫了估计。。。)

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
int const MAXN = 100010;
typedef struct EDGE{
    int to;
    int w;
}Edge;
vector <Edge> edge[MAXN];
queue <int> q;
int cnt[MAXN],vis[MAXN],dis[MAXN];
int father[MAXN];
int n,m;
void Init(){
    for(int i = 0;i < MAXN;i++){
        father[i] = i;
        edge[i].clear();
    }
}
void Add(int u,int v,int w){
    Edge temp;
    temp.w = w;
    temp.to = v;
    edge[u].push_back(temp);
    temp.to = u;
    edge[v].push_back(temp);
}
int Find(int x){
    int temp,y;
    y = x;
    while(y != father[y]){
        y = father[y];
    }
    while(x != father[x]){
        temp = father[x];
        father[x] = y;
        x = temp;
    }
    return y;
}
void Union(int u,int v){
    int uu = Find(u);
    int vv = Find(v);
    if(uu != vv) father[uu] = vv;
}
void Bfs(int start){
    while(!q.empty()){q.pop();};
    memset(vis,0,sizeof(vis));
    memset(dis,-1,sizeof(dis));
    vis[start] = 1;
    dis[start] = 0;
    q.push(start);
    while(!q.empty()){
        int top = q.front();
        q.pop();
        for(int i = 0;i < edge[top].size();i++){
           int to = edge[top][i].to;
           if(!vis[to]){
                vis[to] = 1;
                dis[to] = dis[top] + edge[top][i].w;
                q.push(to);
           }
        }
    }
}
int Count(int start){
    int u = start, maxx = 0;
    Bfs(u);
    for(int i = 1;i <= n;i++){
        if(dis[i] > maxx){
            u = i;
            maxx = dis[i];
        }
    }
    Bfs(u);
    maxx = 0;
    for(int i = 1;i <= n;i++){
        if(dis[i] > maxx) maxx = dis[i];
    }
    return maxx;
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        Init();
        int u,v,w,flag = 0;
        for(int i = 0;i < m;i++){
            scanf("%d%d%d",&u,&v,&w);
            Add(u,v,w);
            if(Find(u) == Find(v)) flag = 1;
            Union(u,v);
        }
        if(flag) printf("YES\n");
        else{
            printf("%d\n",Count(u));
        }
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值