Codeforces1388 D. Captain Flint and Treasure（拓扑排序+处理顺序）

题意：

有n个物品，第i个物品的属性为a(i)和b(i)
现在你要按某种顺序取这些物品，每个物品只能被取一次，最后必须全部取完
如果你取了x，那么总权值增加a(x)，如果b(x)不等于-1，那么a(b(x))+=a(x)

问能取得得最大权值和是多少，并输出取物品的顺序

数据范围：n<=2e5

解法：

因为每个点的出度最多只有一个，容易想到对 i 和b(i)建边，这是一个拓扑图
然后拓扑排序，遍历到i的时候，将i的权值累加给b(i)

但是有一个问题是，当前a(i)有可能是负数，如果累加给b(i)，那么权值和会变小，
因此这时候 i 必须放到最后取

具体实现方法：
1.拓扑排序，如果当前点a(i)>=0，那么累加给b(i)，将i加入答案序列，然后继续正常拓扑排序
如果当前点a(i)<0，那么不累加给b(i)，也不将i加入答案序列，只减少b(i)的入度，将i存到一个栈里面表示最后取。

2.拓扑排序完之后在一次将栈中的数取出，加入答案序列即可。

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=2e5+5;
vector<int>g[maxm];
int in[maxm];
int a[maxm];
int b[maxm];
int n;
vector<int>res;
int ans;
signed main(){
    ios::sync_with_stdio(0);
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=n;i++)cin>>b[i];
    for(int i=1;i<=n;i++){
        if(b[i]!=-1){
            g[i].push_back(b[i]);
            in[b[i]]++;
        }
    }
    queue<int>q;
    stack<int>stk;
    for(int i=1;i<=n;i++){
        if(!in[i]){
            q.push(i);
        }
    }
    while(!q.empty()){
        int x=q.front();q.pop();
        if(a[x]>=0){
            ans+=a[x];
            res.push_back(x);
        }else{//放到最后再取,丢入栈中
            stk.push(x);
        }
        for(int v:g[x]){
            if(a[x]>=0){
                a[v]+=a[x];
            }
            if(in[v]){
                if(--in[v]==0){
                    q.push(v);
                }
            }
        }
    }
    while(!stk.empty()){
        int x=stk.top();stk.pop();
        res.push_back(x);
        ans+=a[x];
    }
    cout<<ans<<endl;
    for(int v:res){
        cout<<v<<' ';
    }
    return 0;
}

---