记录编号 605721 评测结果 AAAAAAAAAAAAAAAAAAAAA
题目名称 4173.[USACO25 Feb Gold]Bessie s Function 最终得分 100
用户昵称 GravatarRuyi 是否通过 通过
代码语言 C++ 运行时间 3.404 s
提交时间 2025-09-06 16:56:33 内存使用 17.59 MiB
显示代码纯文本
#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;
}