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

题意： 编号为 $[1,n]$的点初始权值为 $a_i$，初始答案为 $0$，每次选取一个点 $i$后，答案加上 $a_i$，并且 $i$点不可再次被选，且如果 $a_i\neq1$，给 $a_{b_i}$加上 $a_i$。问可以得到的最大答案和该答案上选择点的顺序。
题解： 典型的拓扑排序，先对每个点计度数。
从度为 $0$的点开始选择：

• 如果点 $i$当前权值小于 $0$，选择点 $i$会减少 $a_{b_i}$，所以暂时不选（形式即为给 $a_{b_i}$加上 $0$）。
• 如果点 $i$的权值不小于 $0$，选择该点不会使 $a_{b_i}$减少，故选择该点。
• 每次枚举完一个点 $i$后，将 $b_i$度数相应减 $1$。

具体实现时可以按照拓扑排序直接写即可。
在输出选择顺序时，先从原始度数最小的点开始，将所有 $a_i$大于等于 $0$的点输出，因为越在前面对后面的点就有影响。然后从原始度数最大的点开始，将所有 $a_i$小于 $0$的点输出，因为越在后面的点先选择，就不会对其后面的点产生负影响。

代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

template<typename T>
inline T Read(){
    T s = 0, f = 1; char ch = getchar();
    while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
    while(isdigit(ch)) {s = (s << 3) + (s << 1) + ch - '0'; ch = getchar();}
    return s * f;
}

#define read() Read<int>()
#define readl() Read<long long>()
const int N = 2e5 + 10;

int n, b[N], d[N];
ll a[N], res;

int p[N], cnt;
int q[N];
void topsort() {
	int hh = 0, tt = -1;
	for(int i = 1; i <= n; ++i) 
		if(!d[i]) q[++tt] = i;
	
	while(hh <= tt) {
		int u = q[hh++];
		p[++cnt] = u;
		res += a[u];
		if(~b[u]) {
			--d[b[u]];
			a[b[u]] += max(0ll, a[u]);
			if(!d[b[u]]) q[++tt] = b[u];
		}
	}
}

void solve() {
	n = read();
	for(int i = 1; i <= n; ++i) d[i] = 0;
	for(int i = 1; i <= n; ++i) a[i] = readl();
	for(int i = 1; i <= n; ++i) {
		b[i] = read();
		if(b[i] == -1) continue;
		++d[b[i]];
	}
	
	topsort();
	
	printf("%lld\n", res);
	vector<int> ans;
	
	for(int i = 1; i <= n; ++i) if(a[p[i]] >= 0) ans.push_back(p[i]);
	for(int i = n; i >= 1; --i) if(a[p[i]] < 0) ans.push_back(p[i]);
	for(int i = 0; i < n; ++i) printf("%d%c", ans[i], " \n"[i == n - 1]);
}

int main()
{
	int T = 1;
	//T = read();
	for(int i = 1; i <= T; ++i) {
		solve();
	}
}