内存分配-拓扑排序

#include <bits/stdc++.h>

using namespace std;

int main()
{
    int n;
    cin >> n;
    vector<int> a(n);
    for(int i = 0; i < n; i ++) cin >> a[i];
    vector<int> g[n];
    vector<int> in(n, 0);
    for(int i = 0; i < n; i ++)
    {
        for(int j = 0; j < n ; j ++)
        {
            int x;
            cin >> x;
            if(x == 1)
            {
                g[j].push_back(i);
                in[i] ++;
            }
        }
    }
    queue<int> q;
    for(int i = 0; i < n; i ++)
    {
        if(!in[i]) q.push(i);
    }
    int res = 0;
    while(!q.empty())
    {
        int size = q.size();
        int cur = 0;
        for(int i = 0; i < size; i ++)
        {
            int u = q.front();
            q.pop();
            cur += a[u];
            for(int v : g[u])
            {
                in[v] --;
                if(!in[v]) q.push(v);
            }
        }
        res = max(res, cur);
    }
    cout << res << endl;
    return 0;
}