[YM]模板-树的直径

概括：

树的直径就是树的一条最长简单路径(不是根的深度)

建议先掌握基础的树形dp

图例：

显然可以观察出3-8(或9)就是树的直径

模板：

dfs两端点：

实质就是做2次dfs找出2个端点

起始任意节点都可以

第二次dfs时用第一次得到的一点端点作为根

注意：有负边权时不能用

/*---code---*/
const int N = 2e5 + 5;
int n;
vector<int>g[N];
int ans,pos;
void dfs(int x,int pr,int d){
    if(d>ans)pos=x,ans=d;
    for(auto u:g[x]){
        if(u==pr)continue;
        dfs(u,x,d+1);
    }
}
void solve(){
    cin>>n;
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,0,0);
    ans=0;
    dfs(pos,0,0);
    cout<<ans;
}

 dp思想：

实质就是找1个节点向下搜索的最长路和次最长路

最长路径=向下搜索的最长路+次最长路

负边权时也能用

/*---code---*/
const int N = 2e5 + 5;
int n;
vector<int>g[N];
int ans;
int dfs(int x,int pr){
    //从x点向下搜索的最长路d1,次最长路d2
    int d1=0,d2=0;
    for(auto u:g[x]){
        if(u==pr)continue;
        int d=dfs(u,x)+1;
        if(d>=d1)d2=d1,d1=d;
        else if(d>d2)d2=d;
    }
    ans=max(ans,d1+d2);
    return d1;
}
void solve(){
    cin>>n;
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,0);
    cout<<ans;
}

典型例题：

Atcoder361:

E - Tree and Hamilton Path 2 (atcoder.jp)

--------------------------------------------------------------------------------------------

题目大意：

求从一个点出发一次性到达所有点的最小路程

题解：

从一个点出发到达所有点的最小距离

其实就是往返的总路程 - 一个单程的最长路

最长路其实就是树的直径

就当于是有一个主干道，你需要去一些支路那个东西再返回主干道，直到到达目的地

Code:

/*---code---*/
const int N = 2e5 + 5;
int n;
vector<pair<ll,ll>>g[N];
ll ans,pos,A;
void dfs(int x,int pr,ll s){
    if(s>ans)ans=s,pos=x;
    for(auto [u,w]:g[x]){
        if(u==pr)continue;
        dfs(u,x,s+w);
    }
}
void solve(){
    cin>>n;
    for(int i=1;i<n;i++){
        ll u,v,w;
        cin>>u>>v>>w;
        A+=w*2;
        g[u].push_back({v,w});
        g[v].push_back({u,w});
    }    
    dfs(1,0,0);
    ans=0;
    dfs(pos,0,0);
    cout<<A-ans;
}