BZOJ1509: [NOI2003]逃學的小孩(樹的直徑)

Time Limit:?5 Sec??Memory Limit:?64 MB
Submit:?1126??Solved:?567
[Submit][Status][Discuss]

Description

Input

第一行是兩個整數N（3 ? N ? 200000）和M，分別表示居住點總數和街道總數。以下M行，每行給出一條街道的信息。第i+1行包含整數Ui、Vi、Ti（1?Ui, Vi ? N，1 ? Ti ? 1000000000），表示街道i連接居住點Ui和Vi，并且經過街道i需花費Ti分鐘。街道信息不會重復給出。

Output

僅包含整數T，即最壞情況下Chris的父母需要花費T分鐘才能找到Chris。

Sample Input

4 3
1 2 1
2 3 1
3 4 1

Sample Output

4

HINT

Source

?

這題比較naive吧。。

不過我一開始以為C是給出的。

很顯然$AB$一定是樹的直徑。

敲完了才發現C是不固定的以為自己白寫了。

但實際上只需要求出直徑的端點到每個點的距離，然后在小的里面取最大就好了

?

#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
#include<map>
#include<cmath>
#include<queue>
#define int long long 
using namespace std;
const int INF = 1e9 + 10, MAXN = 1e6 + 10;
inline int read() {char c = getchar(); int x = 0, f = 1;while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();return x * f;
}
int N, M;
struct Edge {int u, v, w, nxt;
}E[MAXN];
int head[MAXN], num = 1;
inline void AddEdge(int x, int y, int z) {E[num] = (Edge) {x, y, z, head[x]};head[x] = num++;
}
int dis[MAXN], mx, mxdis;
void dfs(int x, int fa) {for(int i = head[x]; i != -1; i = E[i].nxt) {if(E[i].v == fa) continue;dis[E[i].v] = dis[x] + E[i].w;dfs(E[i].v, x);if(dis[E[i].v] > mxdis) mxdis = dis[E[i].v], mx = E[i].v;}
}
int Node1, Node2, dis1[MAXN], dis2[MAXN];
int GetAns() {memset(dis, 0, sizeof(dis)); dfs(Node1, 0);for(int i = 1; i <= N; i++) dis1[i] = dis[i];memset(dis, 0, sizeof(dis)); dfs(Node2, 0);for(int i = 1; i <= N; i++) dis2[i] = dis[i];int rt = 0;for(int i = 1; i <= N; i++)rt = max(rt, min(dis1[i], dis2[i]));return rt;
}
main() { 
#ifdef WIN32freopen("a.in", "r", stdin);
#endifmemset(head, -1, sizeof(head));N = read(); M = read();for(int i = 1; i <= M; i++) {int x = read(), y = read(), z = read();AddEdge(x, y, z);AddEdge(y, x, z);}dfs(1, 0); Node1 = mx;memset(dis, 0, sizeof(dis)); mxdis = 0;dfs(mx, 0); Node2 = mx;int ans = dis[mx];printf("%I64d", ans + GetAns());
}

?