洛谷 P1099 树网的核

题目描述
设T=(V, E, W) 是一个无圈且连通的无向图（也称为无根树），每条边到有正整数的权，我们称T为树网（treebetwork），其中V，E分别表示结点与边的集合，W表示各边长度的集合，并设T有n个结点。
路径：树网中任何两结点a，b都存在唯一的一条简单路径，用d(a, b)表示以a, b为端点的路径的长度，它是该路径上各边长度之和。我们称d(a, b)为a, b两结点间的距离。
　　D(v, P)=min{d(v, u), u为路径P上的结点}。
树网的直径：树网中最长的路径成为树网的直径。对于给定的树网T，直径不一定是唯一的，但可以证明：各直径的中点（不一定恰好是某个结点，可能在某条边的内部）是唯一的，我们称该点为树网的中心。
偏心距ECC(F)：树网T中距路径F最远的结点到路径F的距离，即
ECC(F)=max{d(v, F)，v∈V}
任务：对于给定的树网T=(V, E, W)和非负整数s，求一个路径F，他是某直径上的一段路径（该路径两端均为树网中的结点），其长度不超过s（可以等于s），使偏心距ECC(F)最小。我们称这个路径为树网T=(V, E, W)的核（Core）。必要时，F可以退化为某个结点。一般来说，在上述定义下，核不一定只有一个，但最小偏心距是唯一的。
下面的图给出了树网的一个实例。图中，A-B与A-C是两条直径，长度均为20。点W是树网的中心，EF边的长度为5。如果指定s=11，则树网的核为路径DEFG（也可以取为路径DEF），偏心距为8。如果指定s=0（或s=1、s=2），则树网的核为结点F，偏心距为12。

---

【题目分析】
树形DP+暴力。

---

【代码】

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int h[1001],ne[1001],to[1001],w[1001],en=0;
int n,s,root;
int son[501],dep[501],down1[501],down2[501],tos[501],dist[501];
void add(int a,int b,int c)
{ne[en]=h[a];to[en]=b;w[en]=c;h[a]=en++;}
int bfs(int s)
{
    queue<int>q;
    int dis[501],vis[501];
    memset(dis,0x3f,sizeof dis);
    memset(vis,0,sizeof vis);
    while (!q.empty()) q.pop();
    q.push(s);dis[s]=0;vis[s]=1;
    while (!q.empty())
    {
        int x=q.front();q.pop();
        for (int i=h[x];i>=0;i=ne[i])
        if (!vis[to[i]]){
            dis[to[i]]=dis[x]+w[i];
            q.push(to[i]);
            vis[to[i]]=s;
        }
    }
    int maxx=0,mpos;
    for (int i=1;i<=n;++i)
        if (dis[i]>maxx)
        {
            maxx=dis[i];
            mpos=i;
        }
    return mpos;
}
int f[501];
void dfs(int k)
{
    for (int i=h[k];i>=0;i=ne[i])
    if (to[i]!=f[k]){
        f[to[i]]=k;
        dfs(to[i]);
        if (down1[to[i]]+w[i]>down1[k])
        {
            tos[k]=w[i];
            son[k]=to[i];
            down2[k]=down1[k];
            down1[k]=down1[to[i]]+w[i];
        }
        else if (down1[to[i]]+w[i]>down2[k]) down2[k]=down1[to[i]]+w[i];
    }
    dist[k]=dist[son[k]]+tos[k];
}
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d%d",&n,&s);
    for (int i=1;i<n;++i)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c); add(b,a,c);
    }
    root=bfs(1);
    dfs(root);
    int p=root;
    int hd=1,tl=0;
    int que[501];
    while (p!=0)
    {
        que[++tl]=p;
        p=son[p];
    }
    int tmp=0x3f3f3f3f;
    for (int i=1;i<=tl;++i)
    {
        int ans=0;
        int j=i;
        while (j<tl&&dist[que[i]]-dist[que[j+1]]<=s) j++;
        for (int k=i;k<=j;++k)
            ans=max(ans,down2[que[k]]);
        ans=max(ans,down1[que[j]]);
        ans=max(ans,dist[que[1]]-dist[que[i]]);
        tmp=min(ans,tmp);
    }
    printf("%d\n",tmp);
}