显示代码纯文本
#include<bits/stdc++.h>
#define ll long long
#define N 200001
#define inf 4e18
using namespace std;
ll dp[N][2],f[N][2],c[N];
bool done[N],o[N];
vector<int> out[N],r;
int in[N],n;
void dfs(int u){
done[u]=1;
dp[u][0]=(u!=in[u])*c[u];
for(int v:out[u])
if(!o[v]&&!done[v]){
dfs(v);
dp[u][0]+=min(dp[v][0],dp[v][1]);
dp[u][1]+=dp[v][0];
}
return ;
}
ll deal(int x){
r.clear();
int y=x;
do{
r.push_back(y);
o[y]=1;
y=in[y];
}while(y!=x);
for(int u:r) dfs(u);
int len=r.size();
if(len==1) return min(dp[r[0]][0],dp[r[0]][1]);
for(int i=0;i<len;i++) f[i][0]=f[i][1]=inf;
f[0][0]=dp[r[0]][0];
for(int i=1;i<len;i++){
f[i][0]=min(f[i][0],min(f[i-1][0],f[i-1][1])+dp[r[i]][0]);
f[i][1]=min(f[i][1],f[i-1][0]+dp[r[i]][1]);
}
ll ans=min(f[len-1][0],f[len-1][1]);
for(int i=0;i<len;i++) f[i][0]=f[i][1]=inf;
f[0][0]=dp[r[0]][0];
f[0][1]=dp[r[0]][1];
for(int i=1;i<len;i++){
f[i][0]=min(f[i][0],min(f[i-1][0],f[i-1][1])+dp[r[i]][0]);
f[i][1]=min(f[i][1],f[i-1][0]+dp[r[i]][1]);
}
ans=min(ans,f[len-1][0]);
return ans;
}
int main(){
freopen("Function.in","r",stdin);
freopen("Function.out","w",stdout);
cin>>n;
ll ans=0;
for(int i=1;i<=n;i++)cin>>in[i],out[in[i]].push_back(i);
for(int i=1;i<=n;i++)cin>>c[i];
for(int i=1;i<=n;i++){
if(!done[i]){
int x=in[i],y=i;
while(x!=y)x=in[in[x]],y=in[y];
ans+=deal(x);
}
}
cout<<ans<<endl;
return 0;
}