AcWing 1078. 旅游规划 树型dp（非常详细）

AcWing 1078. 旅游规划

W 市的交通规划出现了重大问题，市政府下定决心在全市各大交通路口安排疏导员来疏导密集的车流。

但由于人员不足，W 市市长决定只在最需要安排人员的路口安排人员。

具体来说，W 市的交通网络十分简单，由 n 个交叉路口和 n−1 条街道构成，交叉路口路口编号依次为 0,1,…,n−1 。

任意一条街道连接两个交叉路口，且任意两个交叉路口间都存在一条路径互相连接。

经过长期调查，结果显示，如果一个交叉路口位于 W 市交通网最长路径上，那么这个路口必定拥挤不堪。

所谓最长路径，定义为某条路径 p=(v1,v2,…,vk)，路径经过的路口各不相同，且城市中不存在长度大于 k 的路径（因此最长路径可能不唯一）。

因此 W 市市长想知道哪些路口位于城市交通网的最长路径上。

输入格式
第一行包含一个整数 n。

之后 n−1 行每行两个整数 u,v，表示编号为 u 和 v 的路口间存在着一条街道。

输出格式
输出包括若干行，每行包括一个整数——某个位于最长路径上的路口编号。

为了确保解唯一，请将所有最长路径上的路口编号按编号顺序由小到大依次输出。

数据范围
1≤n≤2×105
输入样例：

10
0 1
0 2
0 4
0 6
0 7
1 3
2 5
4 8
6 9

输出样例：

0
1
2
3
4
5
6
8
9

这道题是一道树形dp题，其实这道题还是相当有难度的，首先我们要对一个点确定三个变量-$d1[i],d2[i],up[i]$，这三个变量分别代表i的最长长度，i的次长长度，i的向上的延伸的最长长度。我们最后的答案应该是d1,d2,up三者的最大的两个数的和。有一些不是很容易理解的地方我已经标出来了。

首先我们先求出d1,d2操作。
不唯一

y总的方法

void dfs_d(int u,int father)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j!=father)
        {
            dfs_d(j,u);
            int t=d1[j]+1;
            if(t>d1[u])
            d2[u]=d1[u],d1[u]=t,p1[u]=j;
            else if(t>d2[u])
            d2[u]=t;
        }
    }
    maxd=max(maxd,d1[u]+d2[u]);
}

这个思路是向下遍历子节点，我们不会返回父节点，所以可以实现向下遍历。
难点最后我们要求出的最大长度不应该是d1,d2,up三者中最大的两个数的和吗，但是只有d1,d2的和。

其实，我们要考虑有一定有一个点，可以使得它的两个子节点为最大的两项（其实画图应该就可以理解了），不过y总用了这个式子我也是有一点点理解的吃力，最后慢慢理解的。

然后下一个难点是dfs_u操作

void dfs_u(int u,int father)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j!=father)
        {
            up[j]=up[u]+1;
            if(p1[u]==j) up[j]=max(up[j],d2[u]+1);
            else up[j]=max(up[j],d1[u]+1);
            dfs_u(j,u);
        }
    }
    
}

首先我们要对这两行代码详细说一说。

            if(p1[u]==j) up[j]=max(up[j],d2[u]+1);
            else up[j]=max(up[j],d1[u]+1);

这是啥意思呢，这个意思是，如果j在最长长度上，我们就选取d2（因为这个点在d1的线段上，我们不能选择这条啊不然计算结果一定错误），如果不在最长的线段，我们就up[j]选取父节点最长的d1[u]，这个可能需要画图理解一下吧。

可以想一下，选d1这个线段是不是会重合了。

最后一步就是枚举了。

代码如下：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=2e5+10;

int h[N],e[N*2],ne[N*2],idx;
int d1[N],d2[N],p1[N],up[N];
int maxd=0;
int n;

void add(int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}

void dfs_d(int u,int father)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j!=father)
        {
            dfs_d(j,u);
            int t=d1[j]+1;
            if(t>d1[u])
            d2[u]=d1[u],d1[u]=t,p1[u]=j;
            else if(t>d2[u])
            d2[u]=t;
        }
    }
    maxd=max(maxd,d1[u]+d2[u]);
}

void dfs_u(int u,int father)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j!=father)
        {
            up[j]=up[u]+1;
            if(p1[u]==j) up[j]=max(up[j],d2[u]+1);
            else up[j]=max(up[j],d1[u]+1);
            dfs_u(j,u);
        }
    }
    
}

int main(void)
{
    memset(h,-1,sizeof h);
    
    scanf("%d",&n);
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);add(b,a);
    }
    dfs_d(0,-1);
    dfs_u(0,-1);
    for(int i=0;i<n;i++)
    {
        int d[3]={up[i],d1[i],d2[i]};
        sort(d,d+3);
        if(maxd==d[1]+d[2]) cout<<i<<endl;
    }
}