题目链接
多个环,最后一定是剩下中间那条边。因为双方都可以构造出不让对方更优的结果,且两者等价,所以一定是中位数。
但是偶环中位数不唯一,所以贪心地按照两条边和来选取。
#include<bits/stdc++.h>
#define N 100005
typedef long long ll;
using namespace std;
inline ll read(){
ll x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,tot=1,head[N],ver[N<<1],nex[N<<1],edge[N<<1];
struct node{
int u,v;
ll w;
bool operator < (const node &x)const{return w<x.w;}
};
vector<node> cl[N];
int ct,l[N],r[N],fa[N],dep[N],in[N],vi[N];
inline void add(int x,int y,ll z){
nex[++tot]=head[x];head[x]=tot;ver[tot]=y;edge[tot]=z;
}
node tmp;
void work(int u,int v,int c){
if(u==v)return ;
if(dep[u]<dep[v])swap(u,v);
tmp.u=u,tmp.v=fa[u],tmp.w=edge[in[u]];
cl[c].push_back(tmp);
work(fa[u],v,c);
}
void dfs(int x){
vi[x]=1;
for(int i=head[x];i;i=nex[i]){
int y=ver[i];
if(i/2==in[x]/2)continue;
if(vi[y]){
if(dep[x]>dep[y]){
tmp.u=x,tmp.v=y,tmp.w=edge[i];
cl[++ct].push_back(tmp);
work(x,y,ct);
}
}else{
fa[y]=x;dep[y]=dep[x]+1;
in[y]=i;
dfs(y);
}
}
}
int t[N];
ll w[N],ans;
priority_queue<pair<ll, int> > q;
int main(){
int n=read();
for(int i=1;i<=n;++i)t[i]=read();
for(int i=1;i<=n;++i)w[i]=read();
for(int i=1;i<=n;++i){
if(t[i]==i)continue;
add(i,t[i],w[i]);add(t[i],i,w[i]);
ans+=w[i];
}
for(int i=1;i<=n;++i){
if(!vi[i]){
dfs(i);
}
}
for(int i=1;i<=ct;++i){
sort(cl[i].begin(),cl[i].end());
l[i]=(cl[i].size()-1)>>1;r[i]=(cl[i].size())>>1;
if(l[i]==r[i])continue;
q.push(make_pair(cl[i][r[i]].w-cl[i][l[i]].w,i));
}
int op=1;
while(!q.empty()){
int x=q.top().second;q.pop();
if(op)++l[x];
else --r[x];
op^=1;
}
for(int i=1;i<=ct;++i){
ans-=cl[i][l[i]].w;
}
printf("%lld\n",ans);
return 0;
}