链接
https://codeforces.com/contest/1388/problem/D
题意
数组 a , b a,b a,b 长度为 n n n,选择一个 i i i,进行如下操作
- a n s + = a i ans+=a_i ans+=ai
- a b i + = a i ( b i ≠ − 1 ) a_{b_i}+=a_i(b_i\ne-1) abi+=ai(bi=−1)
求一个 1 ∼ n 1\sim n 1∼n 的排列使按此排列操作得到的 a n s ans ans 最大
思路
i i i 对 b i b_i bi 有贡献,即 i i i 可以到达 b i b_i bi,反向建图,我们可以得到若干个树形图,树根为 b i = − 1 b_i=-1 bi=−1 的点
对于某个节点 u u u 的儿子 v v v,若 a v > 0 a_v>0 av>0,贡献为正,则先对 v v v 进行操作,否则 v v v 只能在 u u u 操作后再操作,可以把 v v v 存进数组 t t t
遍历完所有点后,倒序操作数组 t t t 中的所有点,因为后加入的点可能为先加入点的祖先
代码
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
int n,b[N],res[N],cnt;
long long a[N],tot;
vector<int> G[N],t;
void dfs(int u,int fa) {
for(auto v:G[u]) dfs(v,u);
if(a[u]>0) res[++cnt]=u,tot+=a[u],a[fa]+=a[u];
else t.push_back(u);
}
int main() {
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<=n;i++) scanf("%d",&b[i]);
for(int i=1;i<=n;i++) {
if(b[i]==-1) continue;
G[b[i]].push_back(i);
}
for(int i=1;i<=n;i++) if(b[i]==-1) dfs(i,0);
for(int i=t.size()-1;i>=0;i--) res[++cnt]=t[i],tot+=a[t[i]];
printf("%lld\n",tot);
for(int i=1;i<=n;i++) printf("%d ",res[i]);
puts("");
return 0;
}