P1099 [NOIP2007 提高组] 树网的核

P1099 [NOIP2007 提高组] 树网的核

可以采用尺取法的方法去做，先找到树的直径，然后用双指针枚举小于等于s长度的线段，枚举两端点到直径的最大距离的最小值，代码实现如下：

for(int i=top , j=top ; i ;i = fa[i]){
        while(d[j] - d[i] > m && j){
            j = fa[j];
        }
        ans = min(ans,max(d[top]-d[j],d[i]));
}

但此时不能忘记除直径上的点到线段的距离，如下样例：

5 12
1 2 4
2 3 4
3 4 4
2 5 3

如果单考虑如上写法我们得到答案为0，少考虑了除直径上的点，因此我们需要标记，如果是直径上的点，那么就跳过，就相当于将直径缩成一个点。

代码实现：

#include <bits/stdc++.h>
#define int long long
#define fi first 
#define se second
#define all(v) v.begin(),v.end()
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f;
const int N = 2e5+5;
int n,m;
struct node{
    int x,w;
};
vector<node>g[N];
int d[N],fa[N];
int vis[N];
int top,k=0;

void dfs(int x,int p){
    fa[x] = p;
    if(d[x]>d[k])k = x;
    for(const auto &[y,w]:g[x]){
        if(y==p || vis[y])continue;
        d[y] = d[x] + w;
        dfs(y,x);
    }
    return;
}

void solve(){
    cin>>n>>m;
    for(int i=1;i<n;i++){
        int x,y,w;cin>>x>>y>>w;
        g[x].push_back({y,w});
        g[y].push_back({x,w});
    }
    memset(d,0,sizeof(d));
    dfs(1,0);
    d[k] = 0;
    dfs(k,0);

    top = k;
    int ans = inf;
    for(int i=top , j=top ; i ;i = fa[i]){
        while(d[j] - d[i] > m && j){
            j = fa[j];
        }
        ans = min(ans,max(d[top]-d[j],d[i]));
    }

    for(int i=top;i;i=fa[i])vis[i] = 1;

    for(int i=top;i;i=fa[i]){
        d[i] = 0;
        dfs(i,fa[i]);
    }

    for(int i=1;i<=n;i++){
        ans = max(ans,d[i]);
    }

    cout<<ans<<"\n";

    
}

signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int T = 1;
    //cin>>T;
    while(T--){
        solve();
    }
    return 0;
}