树形DP，最大直径，并查集——HDU4514

题目链接

题目含义

建树，如果树里有环就输出YES，如果没有就输出最大直径

题目分析

可以用并查集来判断有没有环，如果当前是树，那么每个节点都在一个集合内

此时树内任意两个节点加上一条边都会成环

接下来的工作就是求最大直径了

这里我是用的DP法求直径，不知道两次DFS行不行

但要注意的是，全部的节点可能在不同的连通分支上

所以可以用一个vis数组来表示这个分支已经被访问过了

题目代码

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int maxn=1e5+7;
int tot,n,m,a,b,c,ans;
int head[maxn],f[maxn],d[maxn];
bool vis[maxn];
struct node{
    int to,w,next;
}edge[maxn*2];
void add(int u,int v,int w){
    edge[tot].to=v;
    edge[tot].w=w;
    edge[tot].next=head[u];
    head[u]=tot++;
}
int get(int x){
    if(f[x]==x)return x;
    else return f[x]=get(f[x]);
}
void join(int x,int y){
    int fx=get(x),fy=get(y);
    f[fx]=fy;
}
void dfs(int u,int fa){
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v==fa||vis[v])continue;
        vis[u]=1;dfs(v,u);
        ans=max(ans,d[u]+d[v]+edge[i].w);
        d[u]=max(d[u],d[v]+edge[i].w);
    }
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        tot=0;
        memset(vis,false,sizeof(vis));
        memset(head,-1,sizeof(head));
        memset(d,0,sizeof(d));
        for(int i=1;i<=n;i++)
            f[i]=i;
        bool flag=false;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&a,&b,&c);
            if(get(a)==get(b)){
                flag=true;
                continue;
            }
            join(a,b);
            add(a,b,c),add(b,a,c);
        }
        if(flag){
            printf("YES\n");
            continue;
        }
        ans=0;
        for(int i=1;i<=n;i++)
        if(!vis[i]){
            dfs(i,0);
            vis[i]=true;
        }
        printf("%d\n",ans);
    }
    return 0;
}