龙龙送外卖[天梯赛]

题目描述

输入样例
7 4
-1 1 1 1 2 2 3
5
6
2
4
输出样例
2
4
4
6

思路

记忆化搜索

我们记sum为需要送外卖的点之间不同的路径，题目要求我们最后不必返回外卖站，我们可以先假设最后返回外卖站，则这样的距离就是sum * 2，最后不返回，我们就减去最后一个外卖店距离外卖站的距离，即为d，这样最后的结果为sum * 2 - d，当d最大时，求解的即时最小的。

采用记忆化搜索，可以防止我们处理重复点时，重复搜索

样例解释

AC代码

#include <iostream>
using namespace std;
const int N = 200010;
int p[N]; //存储每一个点的双亲节点
int d[N]; //存储每一个需要送外卖的点距离外卖站的距离
int sum; //需要送外卖的点之间不同的路径
int n, m;

int dfs(int x) //搜索编号为x的外卖点距离外卖站的距离
{
    if(p[x] == -1 || d[x]) return d[x]; //如果当前这个点已经被访问过，则直接返回其距离即可
    //否则，我们寻找该点距离外卖站的距离，且走过的距离+1
    sum ++;
    d[x] = dfs(p[x]) + 1;
    return d[x];
}
int main()
{
    int max_dis = -1; //在所有需要送外卖的地点中距离外卖站最远的点的距离
    cin >> n >> m;
    for(int i = 1; i <= n; i ++) cin >> p[i];
    while(m --)
    {
        int new_point;
        cin >> new_point;
        max_dis = max(dfs(new_point), max_dis); //求出截止到目前为止，外卖点中距离外卖站最远的距离
        cout << 2 * sum - max_dis << endl;
    }
    return 0;
}

欢迎大家批评指正！！！