【HDU 4514】【树的直径 dfs或者并查集判断环】【给定一个无向图，图可能是非连通的，如果图中存在环，就输出YES，否则就输出树的直径】

传送门：http://acm.split.hdu.edu.cn/showproblem.php?pid=4514

描述：

湫湫系列故事——设计风景线

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 4610    Accepted Submission(s): 840

Problem Description

　　随着杭州西湖的知名度的进一步提升，园林规划专家湫湫希望设计出一条新的经典观光线路，根据老板马小腾的指示，新的风景线最好能建成环形，如果没有条件建成环形，那就建的越长越好。
　　现在已经勘探确定了n个位置可以用来建设，在它们之间也勘探确定了m条可以设计的路线以及他们的长度。请问是否能够建成环形的风景线？如果不能，风景线最长能够达到多少？
　　其中，可以兴建的路线均是双向的，他们之间的长度均大于0。

 

Input

　　测试数据有多组，每组测试数据的第一行有两个数字n, m，其含义参见题目描述；
　　接下去m行，每行3个数字u v w，分别代表这条线路的起点，终点和长度。

　　 [Technical Specification]
　　1. n<=100000 
　　2. m <= 1000000
　　3. 1<= u, v <= n 
　　4. w <= 1000

 

Output

　　对于每组测试数据，如果能够建成环形（并不需要连接上去全部的风景点），那么输出YES，否则输出最长的长度，每组数据输出一行。

 

Sample Input

  
  

   
   3 3
1 2 1
2 3 1
3 1 1
  
  

 

Sample Output

  
  

   
   YES
  
  

 

Source

2013腾讯编程马拉松初赛第二场（3月22日）

 

Recommend

liuyiding   |   We have carefully selected several similar problems for you:   5932  5931  5930  5929  5928 

 

题意：

给定一个无向图，图可能是非连通的，如果图中存在环，就输出YES，否则就输出树的直径

思路：

判断是否成环可以用并查集搞一下或者记录前驱节点dfs一下

为什么会想到这两个呢？最小生成树中判断两顶点是否属于一个连通分量完美的应用了并查集。另外dfs判断环可以参考这道cf 711 D

如何求树的直径呢？
随便找一个点u ，找到距离 u 最远的点 v。
然后再从 v 出发，找距离 v 最远的点 w，路径（v -> w）就是树的直径。

下面两份代码存边和判环的方式都不一样：

代码一：

STL+结构体存边，并查集判环，需要注意的是系统判断内存是总共消耗的内存， 如果找到环了，就不要加边了，不然很容易MLE。

#include <bits/stdc++.h>
using  namespace  std;
#define rep(i,k,n) for(int i=k;i<=n;i++)

const int N=1e5 + 10;

struct node{
  int v, w;
  node(){}
  node(int _v, int _w): v(_v), w(_w) {}
};
std::vector<node> e[N];
int dp[N], p[N], n, m, ans;

int Find(int x){
  return p[x] == x ? x : p[x] = Find(p[x]);
}

void dfs(int u, int fa){
  int mx = 0;
  for(auto es : e[u]){ 
    int v = es.v;
    int w = es.w;
    if(v == fa)continue;
    dfs(v, u);
    ans = max(ans, dp[v] + mx + w);
    mx = max(mx, dp[v] + w);
  }
  dp[u] = mx;
}

void solve(){
  rep(i, 1, n){
    if(dp[i] == -1)
      dfs(i, -1);
  }
  printf("%d\n", ans);
}

int  main(){
  while(~scanf("%d%d", &n, &m)){
    bool flag = 0;
    ans = 0;
    rep(i, 1, n){
      p[i] = i, dp[i] = -1, e[i].clear();
    }
    int u, v, w;
    rep(i, 1, m){
      scanf("%d%d%d", &u, &v, &w);
      if(flag)continue;

      e[u].push_back(node(v, w));//放后面，如果找到环了，就不要加边了，不然很容易MLE
      e[v].push_back(node(u, w));

      int fx = Find(u), fy = Find(v);
      if(fx != fy) p[fy] = fx;
      else flag = true;
    }

    if(flag){
      puts("YES");
      continue;
    }
    solve();
  }
  return 0;
}

代码二：

结构体存边，dfs判环，判环和找树直径在一个dfs里面

#include <bits/stdc++.h>
using  namespace  std;
#define mst(ss,b) memset(ss,b,sizeof(ss));
#define rep(i,k,n) for(int i=k;i<=n;i++)

const int N=1e5 + 10;

struct Edge
{
  int v, w, next;
}e[N * 10 * 2];
int head[N], d[N], tol;
bool vis[N], tmpv[N];
int n, m;

void init(){
  tol = 0;
  mst(head, -1);
}

void addedge(int u, int v, int w){
  e[tol].v = v;
  e[tol].w = w;
  e[tol].next = head[u];
  head[u] = tol ++;
}

bool dfs(int pre, int u){
  vis[u] = true;
  for(int i = head[u]; ~i; i = e[i].next){
    int v = e[i].v;
    if(v == pre)continue;
    if(vis[v])return false;
    d[v] = d[u] + e[i].w;
    if(! dfs(u, v))return false;    
  }
  return true;  
}

int main(){
  while(~scanf("%d%d", &n, &m)){
    init();
    int u, v, w;
    while(m--){
      scanf("%d%d%d", &u, &v, &w);
      addedge(u, v, w);
      addedge(v, u, w);
    }
    int ans = 0;
    bool flag = 0;
    mst(vis, false);
    rep(i, 1, n){
      if(vis[i])continue;
      mst(d, 0);
      if(!dfs(-1, i)){
        flag = true; //找到环了
        break;
      }
      int u = max_element(d + 1, d + n + 1) - d;//dfs一次没找到环，先找出离结点i最远u
      memcpy(tmpv, vis, sizeof(vis));
      mst(vis, 0);
      mst(d, 0);
      dfs(-1, u);//再dfs一次求出u的最远结点就是ans的可能取值
      ans = max(ans, *max_element(d + 1, d + n + 1));
      memcpy(vis, tmpv, sizeof(vis));
    }
    if(flag)puts("YES");
    else printf("%d\n", ans);
  }
  return 0;
}

一道题学到的还挺多的~