Codeforces Round #660 (Div. 2) D. Captain Flint and Treasure ###K ### //K

题目链接：https://codeforces.ml/contest/1388/problem/D

题意：一种操作为 选一个下标 使得ans+=a[i] 且 把a[i]+到a[b[i]]中   要求每个下标都进行一种这样的操作，问怎么样的操作顺序才能使得ans最大

思路：要使得ans最大，那么肯定是a[i]为正数的都尽量早的累加，为负数的都尽量晚的累加，那么现在只需要考虑如何遍历就行了，题目已经说明是

有向无环图，那么首先想到的就应该是拓扑排序， 拓扑排序后， 正数应该从序号小的累加到序号大的， 这样能最大贡献， 负数应该从序号大的累加到序号小的，才能使得负数贡献小

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
const int maxn =2e5+10;
const int mod=1e9+7;
ll a[maxn];
int b[maxn];
int ru[maxn];
int num[maxn];


bool cmp(pair<int,int>a,pair<int,int>b)
{
    return a.first>b.first;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    for(int i=1;i<=n;i++)
    {
        cin>>b[i];
        if(b[i]==-1)
            b[i]=0;
        ru[b[i]]++;
    }
    queue<int>q;
    for(int i=1;i<=n;i++)
    {
        if(ru[i]==0)
            q.push(i);
    }
    ll ans=0;
    vector<int>cnt;
    int tot=0;
    while(!q.empty())
    {
        int u=q.front();
        num[u]=++tot;
        int v=b[u];
        q.pop();
        if(a[u]>=0)
        {
            a[v]+=a[u];
            ans+=a[u];
            cnt.pb(u);
        }
        ru[v]--;
        if(ru[v]==0&&v!=0)
            q.push(v);
    }
    vector<pair<int,int>>temp;
    for(int i=1;i<=n;i++)
    {
        if(a[i]<0)
        {
            temp.pb({num[i],i});
        }
    }
    sort(temp.begin(),temp.end(),cmp);
    for(auto &v:temp)
    {
        ans+=a[v.second];
        cnt.pb(v.second);
    }

    cout<<ans<<'\n';
    for(auto &v:cnt)
        cout<<v<<" ";





}