旅游规划(树的直径树形dp做法)

1078. 旅游规划

旅游规划

思路

1、先通过树形dp求出每个点往下走的最大长度和次大长度，并且更新整棵树的最大路径
2、枚举每一个结点i，设该点的最大长度是d1[i]，次大长度是d2[i]，若d1[i] + d2[i] ==ans，表示最大路径是从该点得出的，通过dfs2找到最大长度的路径，dfs3找到次大长度的路径，并对路径的点进行标记

样例输入：

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

代码：

//树的中心 --> 树的直径
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;

const int N = 2e5 + 10, M = N * 2;
int n;
int h[N], ne[M], e[M], idx;
int d1[N], d2[N], p[N], up[N];
//d1向递归的最长距离 d2向下递归的次长距离 p为向下递归的路径 up为向上的最长距离
int maxx;
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void dfs_d(int u, int fa)
{
    for(int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if(j == fa) continue;
        dfs_d(j, u);
        int d = d1[j] + 1;
        if(d > d1[u])
        {
            d2[u] = d1[u], d1[u] = d;
            p[u] = j;
        }
        else if(d > d2[u]) d2[u] = d;
    }
    
    maxx = max(maxx, d1[u] + d2[u]);
    return;
}

void dfs_u(int u, int fa)
{
    for(int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if(j == fa) continue;
        up[j] = up[u] + 1;
        if(j == p[u]) 
        {
            //如果当前的路径在down的最长距离路径上,up即为down的次长距离或者dfs根节点的距离
            up[j] = max(d2[u] + 1, up[j]);
        }else        
        {
            //如果当前的路径不在down的最长路径上, up即为down的最长距离或者dfs根节点的距离
            up[j] = max(d1[u] + 1, up[j]);
        }
        
        dfs_u(j, u);
    }
    
    return;
}

int main()
{
    cin >> n;
    memset(h, -1, sizeof h);
    for(int i = 1; i < n; i ++ )
    {
        int a, b;
        cin >> a >> b;
        add(a, b), add(b, a);  //无向图
    }
    
    dfs_d(0, -1);
    dfs_u(0, -1);
    
    for(int i = 0; i < n; i ++ )
    {
        int s[3] = {d1[i], d2[i], up[i]};
        sort(s, s + 3);
        if(s[1] + s[2] == maxx) cout << i << endl;
    }
    
    return 0;
    
}

---