2019 Multi-University Training Contest 3 1011 Squrirrel —— 树形DP

This way

题意：

给你一棵树，路径上都有权值，你现在能将一条边权变为0，问你从某个节点出发到最远的叶子结点的路径最短是多少。

题解:

这道题乍一看不难，但是它的情况很多，所以debug了很久才过。首先我们处理出每个节点距离它子树的叶子结点的最远，第二远，第三远的距离，再用DP做出每个节点删掉一条边后到叶子结点最远，第二远的距离。之后再用一个dfs做答案，首先参数需要传：
1.从父亲传下来的，不包括这颗子树的最远未删边距离，2.从父亲传下来的，不包括这颗子树的最优最远删边距离。
那么这个节点的答案就是
min{max{父亲未删边,这棵树删边最大，这棵树未删边第二大},max{父亲删边最优最大，这棵树未删边最大}}
为什么，因为不确定删边最大和未删边第二大的关系，所以需要判一下。
先说一下重要数组的意义：
no_mx：表示未删边最大
no_smx：表示未删边第二大
del_mx：表示删边最大
no_：表示父亲传下来的未删边最大
del_：表示父亲传下来的删边最大
那么之后麻烦的就来了，首先对于他的某一个儿子节点，
1.如果他是这个点的未删边最大的儿子节点：
1.1如果他是这个点的删边最大儿子节点：
1.1.1他就会有no_smx,no_tmx,del_smx,del_tmx四种可以传下去的，那么我们肯定是删掉no_smx,那么也就是说在no_tmx和del_smx中选择最大的
1.2如果他是这个点的第二或者其它儿子节点：
1.2.1他就会有no_smx,no_tmx,del_mx,del_tmx四种可以传下去的，那么我们不知道删掉no_smx还是del_mx，那么我们就在这两个中取一个最小的，在之后与no_tmx取个最大的
2.如果他是这个节点的未删边第二：
2.1如果它是这个点的删边最大儿子节点：
2.1.1他就会有no_mx,no_tmx,del_mx,del_tmx，那么我们肯定是删掉no_mx，所以在del_smx与no_tmx中取个最大的
2.2如果它是这个节点的删边第二大儿子节点或者其它：
2.2.1它就会有no_mx,no_tmx,del_mx,del_tmx四种，那么我们删掉no_mx，所以在del_mx,与no_tmx中取个最大的
3.如果它是这个节点的未删边第三或者其它节点：
3.1如果它是这个点的删边最大儿子节点：
3.1.1它就会有no_mx，no_smx,del_smx,del_tmx四种，那么我们删掉no_mx,在no_smx,del_smx中取最大的
3.2如果它是这个点的删边第二或者其它儿子节点：
3.2.1它就会有no_mx,no_smx,del_mx,del_tmx四种，那么我们删掉no_mx,在no_smx,del_mx中取最大的

情况数有点多，令人绝望的代码。

#include<bits/stdc++.h>
using namespace std;
const int N=2e6+5;
int dp[N][2],no_mx[N],no_smx[N],no_tmx[N],del_mx[N],del_smx[N],del_tmx[N],ans[N];
struct node
{
    int to,next,w;
}e[N*2];
int cnt,head[N];
void add(int x,int y,int w)
{
    e[cnt].to=y;
    e[cnt].next=head[x];
    e[cnt].w=w;
    head[x]=cnt++;
}
void dfs(int x,int fa)
{
    for(int i=head[x];~i;i=e[i].next)
    {
        int ne=e[i].to,w=e[i].w;
        if(ne==fa)
            continue;
        dfs(ne,x);
        if(no_mx[ne]+w>no_mx[x])
            no_tmx[x]=no_smx[x],no_smx[x]=no_mx[x],no_mx[x]=no_mx[ne]+w;
        else if(no_mx[ne]+w>no_smx[x])
            no_tmx[x]=no_smx[x],no_smx[x]=no_mx[ne]+w;
        else if(no_mx[ne]+w>no_tmx[x])
            no_tmx[x]=no_mx[ne]+w;


        int d=min(dp[ne][1]+w,dp[ne][0]);
        dp[x][1]=min(max(dp[x][0],d),max(dp[x][1],dp[ne][0]+w));
        dp[x][0]=max(dp[x][0],dp[ne][0]+w);

        if(d>del_mx[x])
            del_tmx[x]=del_smx[x],del_smx[x]=del_mx[x],del_mx[x]=d;
        else if(d>del_smx[x])
            del_tmx[x]=del_smx[x],del_smx[x]=d;
        else if(d>del_tmx[x])
            del_tmx[x]=d;
    }
}
int p,val;
void fans(int x,int fa,int no_,int del_)
{
    ans[x]=min(max(no_,max(del_mx[x],no_smx[x])),max(del_,no_mx[x]));
    if(ans[x]<val)
        val=ans[x],p=x;
    else if(ans[x]==val&&p>x)
        p=x;
    for(int i=head[x];~i;i=e[i].next)
    {
        int ne=e[i].to,w=e[i].w;
        if(ne==fa)
            continue;
        int nv=no_+w,nd;
        if(no_mx[ne]+w==no_mx[x])
        {
            nv=max(nv,no_smx[x]+w);
            nd=min(max(no_,no_smx[x]),max(del_+w,no_smx[x]+w));//去掉这条边，父亲传来的值
            if(min(dp[ne][1]+w,dp[ne][0])==del_mx[x])
                nd=min(nd,max(del_smx[x]+w,max(no_+w,no_tmx[x]+w)));//去掉其它儿子边
            else
                nd=min(nd,max(no_+w,max(no_tmx[x]+w,min(del_mx[x]+w,no_smx[x]+w))));

        }
        else if(no_mx[ne]+w==no_smx[x])
        {
            nv=max(nv,no_mx[x]+w);
            nd=min(max(no_,no_mx[x]),max(del_+w,no_mx[x]+w));
            if(min(dp[ne][1]+w,dp[ne][0])==del_mx[x])
                nd=min(nd,max(del_smx[x]+w,max(no_+w,no_tmx[x]+w)));
            else
                nd=min(nd,max(no_+w,max(no_tmx[x]+w,del_mx[x]+w)));
        }
        else
        {
            nv=max(nv,no_mx[x]+w);
            nd=min(max(no_,no_mx[x]),max(del_+w,no_mx[x]+w));
            if(min(dp[ne][1]+w,dp[ne][0])==del_mx[x])
                nd=min(nd,max(no_+w,max(no_smx[x]+w,del_smx[x]+w)));
            else
                nd=min(nd,max(no_+w,max(no_smx[x]+w,del_mx[x]+w)));
        }
        fans(ne,x,nv,nd);
    }
}
int main()
{
    //freopen("1.in","r",stdin);
    //freopen("out.txt","w",stdout);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        scanf("%d",&n);

        memset(head,-1,sizeof(head));
        cnt=0;
        val=1e9;
        for(int i=1;i<=n;i++)
            dp[i][0]=dp[i][1]=no_mx[i]=no_smx[i]=no_tmx[i]=del_mx[i]=del_smx[i]=del_tmx[i]=ans[i]=0;

        int x,y,w;
        for(int i=1;i<n;i++)
            scanf("%d%d%d",&x,&y,&w),add(x,y,w),add(y,x,w);
        dfs(1,0);
        fans(1,0,0,0);
        printf("%d %d\n",p,val);
    }
    return 0;
}
/*
1
5
1 2 1
2 3 1
3 4 5
3 5 1

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

1
10
2 1 155
3 1 29
4 2 33
5 2 178
6 3 42
7 2 56
8 6 189
9 4 99
10 5 63
*/