Codeforces Round #660 (Div. 2) D

由题可以构造出一个有向路径，对于一个路径，只要我们从起点开始走，我们就可以尽可能的得到最大的值。
若前面一个是负数，则我们把它最后选取最为合适，因为这样继续跟着这条路径选下去，会导致后面的数越变越小。

直接拓扑排序正数存到res1,负数存到res2.
对于res2，为了自己的贪心成立，对于每一个负数，可能会有它前面的一个点是它的父父父结点，我们不能让这种情况发生，因为发生了，这个数还会变的更小，所以我们直接逆序输出即可。

代码：

#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define int long long
#define sc scanf
#define pf printf
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
typedef unsigned long long LL;
 
const int INF = 0x3f3f3f3f;
const double eps = 1e-4;
const int mod = 1e9+7;
const int N = 200010;

int a[N],b[N],d[N],n;
int e[N],ne[N],h[N],idx=1;
vector<int> res1,res2;

void add(int a,int b){
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs_add(int u){
	if(b[u] == -1 || e[h[u]] == b[u]) return ;
	add(u,b[u]);
	d[b[u]] ++;
	dfs_add(b[u]);
}

void top_sort(){
	int ans = 0;
	queue<int> qe;
	for(int i=1;i<=n;i++)
		if(!d[i]) qe.push(i);
	while(!qe.empty()){
		int t = qe.front(); qe.pop();
		if(t == 0) continue;
		ans += a[t];
		int j = e[h[t]];
		if(--d[j] == 0) qe.push(j);
		if(a[t] > 0) a[j] += a[t];
		if(a[t] > 0) res1.push_back(t);
		else res2.push_back(t);
	}
	cout<<ans<<endl;
}

signed main(){
//	IOS;
	#ifdef ddgo
		freopen("C:/Users/asus/Desktop/ddgoin.txt","r",stdin);
	#endif
	
	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(!d[i]) dfs_add(i);
	
	top_sort();
	for(int i=0;i<(int)res1.size();i++) cout<<res1[i]<<" ";
	for(int i=(int)res2.size()-1;i>=0;i--) cout<<res2[i]<<" ";
	puts("");
	
	return 0;
}