正向dfs+记忆化 20pts+MLE求助
查看原帖
正向dfs+记忆化 20pts+MLE求助
403978
wuzhineihan楼主2024/9/7 19:16

代码如下

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int dfs(vector<vector<int>> &adjacent, int root, vector<int>& memory)
{
    if (memory[root])
    {
        return memory[root];
    }
    else
    {
        int maxn = root;
        for (int neighbor : adjacent[root])
        {
            maxn = max(maxn, dfs(adjacent, neighbor, memory));
        }
        memory[root]=maxn;
        return maxn;
    }
}

int main()
{
    // graph initialization
    int n, m;
    cin >> n >> m;
    vector<vector<int>> adjacent=vector<vector<int>>(n+1);
    int x, y;
    for (int i = 0; i < m; i++)
    {
        cin >> x >> y;
        adjacent[x].push_back(y);
    }

    vector<int> memory=vector<int>(n+1);
    for (int i = 1; i <= n; i++)
    {
        sort(adjacent[i].begin(),adjacent[i].end());
        dfs(adjacent, i, memory);
        cout<<memory[i]<<" ";
    }
}

https://www.luogu.com.cn/record/176195938

2024/9/7 19:16
加载中...