Terrorist’s destroy HDU - 4679(树形dp+树的直径的应用)

题意：（题意来源 https://blog.csdn.net/ayecsz/article/details/10046759）

给你一棵树和边上的权值，定义去掉一条边的花费为边权值（a）乘上b，b定义为去掉边后形成的两棵树中两点间的最远距离（注意是各自内部的最远距离），问去掉哪条边的花费最少？

分析：首先我们可以将树的直径找出来，如下图

假设1->6就是这棵树的直径，非直径上的边为枝条，有两种删除边的方式：

1.去掉的是枝条

2.去掉的是直径上的边

思路：因为每条边的长度都是一，所以我们两遍dfs（分别从1点和第一次dfs后深度最大的点）两次dfs后可以得到树的直径并且将直径上的点做上标记。

处理后，我们还需要解决两个问题，一个是边的编号问题，这个在前向星里加个id属性就行了，

还有我们需要知道一每个点为根的子树的直径（题目问的两边最大的，那么我们从直径的两个端点分别进行dfs就行了），求子树的直径的状态转移有点难以理解，程序里，设置了三个数组，mm[u]（记录直径），lm[u]记录最长子树，rm[u]记录次长子树，因为以u为根的子树的直径，必然通过u，所以只需要找到它的最长，次长子树直径，相加就行了。

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000") //加栈
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <map>
#include <vector>
#include <set>
#include <string>
#include <math.h>
#include <stack>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
#define inf 0x3f3f3f3f
int n;
int num,cnt;
int head[maxn];
int depth[maxn];
int a[maxn];
bool vis[maxn];
int father[maxn];
int mm[maxn],lm[maxn],rm[maxn];
struct Edge
{
    int u,v,next,w,id;
}edge[maxn<<1];
void addEdge(int u,int v,int w,int id)
{
    edge[num].u=u;
    edge[num].v=v;
    edge[num].w=w;
    edge[num].id=id;
    edge[num].next=head[u];
    head[u]=num++;
}
void init()
{
    num=0;
    memset(head,-1,sizeof(head));
}
void dfs1(int u,int pre)
{
    father[u]=pre;
    depth[u]=depth[pre]+1;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v,w=edge[i].w;
        if(v==pre) continue;
        dfs1(v,u);
    }
}
void dfs(int u,int pre)
{
    mm[u]=0,lm[u]=0,rm[u]=0;//记录最长和次长
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        if(v==pre) continue;
        dfs(v,u);
        if(lm[v]+1>rm[u])
        {
            rm[u]=lm[v]+1;
            if(rm[u]>lm[u])
            {
                swap(rm[u],lm[u]);
            }
        }
        if(mm[v]>mm[u])
        {
            mm[u]=mm[v];
        }
    }
    mm[u]=max(mm[u],lm[u]+rm[u]);
}
void solve(int u,int pre)
{
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v,w=edge[i].w,id=edge[i].id;
        if(v==pre) continue;
        solve(v,u);
        if(vis[v])//在直径上
        {
            a[id]=max(a[id],w*mm[v]);
        }
        else
        {
            a[id]=max(a[id],w*cnt);
        }
    }
}
int main(int argc, char const *argv[])
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
        freopen("out.txt","w",stdout);
    #endif
    int T;
    cin>>T;
    int Case=0;
    while(T--)
    {
        init();
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            int x,y,w;
            scanf("%d%d%d",&x,&y,&w);
            addEdge(x,y,w,i);
            addEdge(y,x,w,i);
        }
        father[0]=0;
        dfs1(1,0);
        int u=1;
        for(int i=1;i<=n;i++)
        {
            if(depth[i]>depth[u])
            {
                u=i;
            }
        }
        dfs1(u,0);//两次dfs即可确定直径
        int v=1;
        for(int i=1;i<=n;i++)
        {
            if(depth[i]>depth[v])
            {
                v=i;
            }
        }
        cnt=depth[v]-1;//直径上的长度
        memset(vis,0,sizeof(vis));
        int tmp=v;
        while(tmp)
        {
            vis[tmp]=1;//标记直径上的路径
            tmp=father[tmp];
        }
        for(int i=1;i<=n;i++) a[i]=0;

        dfs(u,0);//正反两次dfs，确定最长子树直径
        solve(u,0);

        dfs(v,0);
        solve(v,0);
        int ans=1e9+7,k=0;
        for(int i=1;i<n;i++)
        {
            if(a[i]<ans)
            {
                ans=a[i];
                k=i;
            }
        }
        printf("Case #%d: %d\n",++Case,k);
    }
    return 0;
}