hdu 4514 湫湫系列故事——设计风景线(dfs)

来源：互联网发布：日本杏林大学知乎编辑：程序博客网时间：2024/05/20 03:46

小记：最开始读不懂题，然后看了下discuss里，原来是求判环，若没环则求森林里的树的最长直径（后来才知道一棵树里距离最远的两点的距离叫树的直径），看了之后，最开始想到了并查集判环，但是后面想不到如果求树的最长直径了，于是沉思良久，dfs可以做到.

思路：对于dfs的某一点，我们保存它两个数据，因为是一棵树，所以对于其中一个点，以它为起点，dfs的某两个终点，通过这个起点，这一条路径就是以该起点为根节点的树的最长直径，这样dfs所有节点，最后得到的最长直径就是answer了.搜i时，以i为根节点的树，我们相当于只要保存深搜以它为起点的路径中长度最长的前两个即是以i为根节点的树的最长直径了.

代码：(len[i] 以i为根节点的树的最长直径，leni[i] 以i为根节点且也以i为起点的最长深搜路径)

#pragma comment(linker, "/STACK:102400000,102400000")

//上面这一句 在HDU交C++的话 要加上去，  交GNU++的话 可以不用 #include <string.h>#include <stdio.h>#define MAX_ 100010#define MAX 1000010#define MIN -0xfffffffstruct point {    int v,cap,next;} edge[MAX*2];int head[MAX_];bool  vis[MAX_];int len[MAX_], leni[MAX_];int ans, M;void add(int from, int to ,int cap) {    //edge[M].u = from;    edge[M].v = to;    edge[M].cap = cap;    edge[M].next = head[from];    head[from] = M++;    //edge[M].u = to;    edge[M].v = from;    edge[M].cap = cap;    edge[M].next = head[to];    head[to] = M++;}bool dfs0(int u,int pre){    int v;    vis[u] = 1;    for(int i = head[u]; i != -1; i = edge[i].next){        v = edge[i].v;        if(v == pre)continue;        if(vis[v] || !dfs0(v,u) )return 0;    }    return 1;}void dfs(int k, int pre) {    int i, v, tmp;    leni[k] = len[k] = 0;    vis[k] = 0;    for(i = head[k]; i != -1; i = edge[i].next) {        v = edge[i].v;        if(pre == v)continue;        dfs(v,k);        tmp = leni[v] + edge[i].cap + leni[k];        if(tmp > len[k]){            len[k] = tmp;        }        if(leni[v] + edge[i].cap > leni[k]){                leni[k] = leni[v] + edge[i].cap;        }    }    if(len[k] > ans)ans = len[k];}int main() {    int m, i, n;    int s,t,c;        while(~scanf("%d%d",&m,&n)) {        M = 0;        memset(head,-1,sizeof(head));        memset(vis,0,sizeof(vis));        for(i = 0; i < n; i++) {            scanf("%d%d%d",&s,&t,&c);            add(s,t,c);        }        for(i = 1; i <= m; i++) {            if(!vis[i]&&!dfs0(i,-1))break;        }        if(i <= m){            printf("YES\n");        }        else {            ans = 0;            for(i = 1; i <= m; ++i){                if(vis[i]){                    dfs(i,-1);                }            }            printf("%d\n",ans);        }    }    return 0;}

0 1