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 + , MAXN = 1e6 + ;
inline int read() {
    char c = getchar(); int x = , f = ;
    while(c < '' || c > '') {if(c == '-') f = -; c = getchar();}
    while(c >= '' && c <= '') x = x *  + c - '', c = getchar();
    return x * f;
}
int N, M;
struct Edge {
    int u, v, w, nxt;
}E[MAXN];
int head[MAXN], num = ;
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 != -; 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, , sizeof(dis)); dfs(Node1, );
    for(int i = ; i <= N; i++) dis1[i] = dis[i];
    memset(dis, , sizeof(dis)); dfs(Node2, );
    for(int i = ; i <= N; i++) dis2[i] = dis[i];
    int rt = ;
    for(int i = ; i <= N; i++)
        rt = max(rt, min(dis1[i], dis2[i]));
    return rt;
}
main() {
#ifdef WIN32
    freopen("a.in", "r", stdin);
#endif
    memset(head, -, sizeof(head));
    N = read(); M = read();
    for(int i = ; i <= M; i++) {
        int x = read(), y = read(), z = read();
        AddEdge(x, y, z);
        AddEdge(y, x, z);
    }
    dfs(, ); Node1 = mx;
    memset(dis, , sizeof(dis)); mxdis = ;
    dfs(mx, ); Node2 = mx;
    int ans = dis[mx];
    printf("%I64d", ans + GetAns());
}