当时没写出来,感觉自己还是太菜了,思维没碰上边
仔细读题,题目说b不存在循环,总是以-1结束
这就意味这开始,肯定有很多点不会受其他点的影响
拓 扑 排 序 ! ! 拓扑排序!! 拓扑排序!!
让 i 到 b i 连 一 条 边 , 意 味 着 如 果 选 了 i , 那 么 a b i 会 加 上 a i 让i到b_i连一条边,意味着如果选了i,那么a_{b_i}会加上a_i 让i到bi连一条边,意味着如果选了i,那么abi会加上ai
拓 扑 排 序 , 如 果 当 前 的 a i 大 于 0 肯 定 让 它 传 递 下 去 , 加 入 到 答 案 序 列 去 拓扑排序,如果当前的a_i大于0肯定让它传递下去,加入到答案序列去 拓扑排序,如果当前的ai大于0肯定让它传递下去,加入到答案序列去
如 果 小 于 0 , 就 不 应 该 让 它 传 递 下 去 , 但 入 度 还 是 一 样 减 , 加 入 到 临 时 序 列 去 如果小于0,就不应该让它传递下去,但入度还是一样减,加入到临时序列去 如果小于0,就不应该让它传递下去,但入度还是一样减,加入到临时序列去
最 后 正 序 输 出 答 案 序 列 , 逆 序 输 出 临 时 序 列 . 最后正序输出答案序列,逆序输出临时序列. 最后正序输出答案序列,逆序输出临时序列.
为 啥 逆 序 输 出 临 时 序 列 ? 因 为 后 进 临 时 序 列 的 一 定 是 在 图 中 深 度 比 较 深 的 为啥逆序输出临时序列?因为后进临时序列的一定是在图中深度比较深的 为啥逆序输出临时序列?因为后进临时序列的一定是在图中深度比较深的
如 果 先 选 这 个 点 , 可 能 对 之 前 的 进 入 临 时 序 列 造 成 影 响 , 所 以 逆 序 。 如果先选这个点,可能对之前的进入临时序列造成影响,所以逆序。 如果先选这个点,可能对之前的进入临时序列造成影响,所以逆序。
#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
typedef long long ll;
ll b[maxn],a[maxn],n,in[maxn],ans;
vector<int>vec[maxn],res,fu;
queue<int>q;
int main()
{
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 ) in[ b[i] ]++,vec[i].push_back( b[i] );
}
for(int i=1;i<=n;i++)
if( in[i]==0 ) q.push(i);
while( !q.empty() )
{
int u=q.front(); q.pop();
ans+=a[u];
if( a[u]>=0 ) res.push_back(u);
else fu.push_back(u);
for(int i=0;i<vec[u].size();i++)
{
int v=vec[u][i];
if( a[u]>=0 ) a[v]+=a[u];
if( --in[v]==0 ) q.push(v);
}
}
cout << ans << endl;
for(int i=0;i<res.size();i++) cout << res[i] << " ";
for(int i=fu.size()-1;i>=0;i--) cout << fu[i] << " ";
}