【DFS好题】BZOJ1999- [Noip2007]Core树网的核（数据加强版）

NOIP的数据好水，一开始有好几个错结果NOIP数据就水过了？？

【题目大意】

求无根树的直径上一段不超过S长的链，使得偏心距最小。具体概念见原题。

【思路】

首先明确几个性质：

（1）对于树中的任意一点，距离其最远的点一定是树的直径的某一端点。

（2）所有的直径是等价的，即任意一条所能求出的该最小偏心距相等。

于是我们可以用两次dfs求出直径。任取一个点找到离它最远的点r，再从r找到距离它最远的点l。l到r的路径就是直径。

显然在长度不超过S的情况下，链最长最好。在l到r上维护尽可能长的链，找到左右端点到直径做右端点的较大值的最小值。然后由链上各个点出发，找到不经过直径上的点抵达的其他点的最大深度。这个最大深度和之前的最小值中较大的就是答案。

为什么是整条直径上找最大深度，而不是在核上找呢？

显然，如果这个深度最深的点不是从核中的点，那么它到核的距离必定小于核的端点到直径端点的距离。所以如果有个节点到核的距离小于核的端点到直径端点的距离，那么它必定是从核上延伸出去的。

【错误点】

具体见程序。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=500000+50;
const int INF=0x7fffffff;
struct edge
{
    int to,len;
};
vector<edge> E[MAXN];
int n,s;
int l,r,dis[MAXN],f[MAXN],ban[MAXN];
 
void addedge(int u,int v,int w)
{
    E[u].push_back((edge){v,w});
    E[v].push_back((edge){u,w});
}
 
void init()
{
    scanf("%d%d",&n,&s);
    for (int i=1;i<n;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        addedge(u,v,w);
    }
}
 
void dfs(int u,int fa)
{
    f[u]=fa;
    for (int i=0;i<E[u].size();i++)
    {
        int to=E[u][i].to;
        if (ban[to] || to==f[u]) continue;
        dis[to]=dis[u]+E[u][i].len;
        dfs(to,u);
    }
}
 
void getd()
{
    memset(ban,0,sizeof(ban));
     
    l=1,r=1;
    dfs(l,0);
    for (int i=1;i<=n;i++) if (dis[i]>dis[r]) r=i;
     
    l=r;
    dis[r]=0;
    dfs(r,0);
    for (int i=1;i<=n;i++) if (dis[i]>dis[l]) l=i;
}
 
void solve()
{
    int i=l,j=l,ans=INF;
    for (;i;i=f[i])
    {
        while (f[j] && dis[i]-dis[f[j]]<=s) j=f[j];
        ans=min(ans,max(dis[j],dis[l]-dis[i]));
        //每次找到以i为一个端点的最接近于S的链
        //比较两端点和直径端点的长度 
    }
    for (i=l;i;i=f[i]) ban[i]=1;
    //由于要找出不经过直径的最大深度，所以禁止访问直径上的点 
    for (int i=l;i;i=f[i]) dis[i]=0,dfs(i,f[i]);
    //★★★★★★★注意这里i的父亲必须传进去f[i],否则就修改了直径 
    for (int i=1;i<=n;i++) ans=max(ans,dis[i]);
    printf("%d",ans);
}
 
int main()
{
    init();
    getd();
    solve();
    return 0;
}