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

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

Time Limit: 6000/3000 MS (Java/Others) Memory Limit: 65535/32768 K (Java/Others)

Total Submission(s): 5769 Accepted Submission(s): 1056

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

题解:判环+求树的直径，本来是用并查集判环，但内存一直超就直接dfs了，树形dp求输出直径，g[u]表示往儿子方向走的最长路，f[u]表示第一步往父亲方向走的最长路=max(f[pre[u]], g[v=pre->v]+w(pre,v)] + w(u->pre);v!=u && v != pre[pre].

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define INF 0x3f3f3f3f
#define N 100001
struct node {
int v;
int w;
int next;
}edge[N*20];
int n, m;
//int father
;
int head
, cnt;
int f
, g
, p
;
bool vis
;
void addedge(int u, int v, int w)
{
edge[++cnt].v = v;
edge[cnt].w = w;
edge[cnt].next = head[u];
head[u] = cnt;
}
/*
int find_father(int x)
{
int a = x;
while(x != father[x])
x = father[x];

while(a != father[a]) {
int z = a;
a = father[a];
father[z] = x;
}
return x;
}*/
bool dfs(int u, int pre)
{
vis[u] = true;
for(int i = head[u]; ~i; i =edge[i].next) {
int v = edge[i].v;
if(v == pre)
continue;
if(vis[v]||dfs(v, u))   return true;
}
return false;
}
void dfs1(int pos, int pre)
{
vis[pos] = true;
g[pos] = 0;
for(int i = head[pos]; ~i; i = edge[i].next) {
int v = edge[i].v;
if(v == pre)
continue;
dfs1(v, p[v]=pos);
g[pos] = max(g[pos], g[v]+edge[i].w);
}
}
void dfs2(int pos, int pre)
{
int t = 0;
f[pos] = f[pre];
for(int i = head[pre]; ~i; i = edge[i].next) {
int v = edge[i].v;
if(v == p[pre])
continue;
if(v == pos)
t = edge[i].w;
else
f[pos] = max(f[pos], g[v]+edge[i].w);
}
f[pos] += t;
for(int i = head[pos]; ~i; i = edge[i].next) {
int v = edge[i].v;
if(v != pre)
dfs2(v, pos);
}
}
int main(int argc, char const *argv[])
{
while(scanf("%d%d", &n, &m) != EOF) {
memset(head, -1, sizeof(head));
for(int i = 0; i <= n; i++)
vis[i] = false;
int u, v, val;
cnt = 0;
bool flag = false;
for(int i = 1; i <= m; i++) {
scanf("%d%d%d", &u, &v, &val);

addedge(u, v, val);
addedge(v, u, val);
/*
int faA = find_father(u);
int faB = find_father(v);
if(faA == faB) {
flag = true;
} else {
father[u] = v;
}
*/
}
for(int i = 1; i <= n; i++) {
if(!vis[i]) {
if(dfs(i, 0)) {
flag = true;
break;
}
}
}
memset(vis, false, sizeof(vis));
if(flag)
printf("YES\n");
else {
for(int i = 1; i <= n;i ++) {
if(!vis[i]) {
dfs1(i, 0);
dfs2(i, 0);
}
}
int ans = -INF;
for(int i = 1; i <= n; i++) {
ans = max(ans, max(g[i], f[i]));
}
printf("%d\n", ans);
}
}
return 0;
}