【洛谷 P7528】 【并查集】 【DFS】 Portals G
题目
解题思路
题目出得很迷惑,掩人耳目啊
题目大意
有2n个点
每次给出5个数,c,p1,p2,p3,p4,p1与p2连边,p3与p4连边,c是换边的花费 例如p1本来和p2连,现在想p1与p3连,p2与p4连,那么花费是c
求要使这2n个点互相连通,最小的花费
看到题目里加粗的那行字,每个点只会出现两次,那么只会有2条边和它相连
一个点能到达的点一定是一个环
现在就是要花费ci使各个环相通
花费按从小到大排序
因为p1与p2是一个环,p3与p4是一个环,如果所属不同的环,那就需要花费且合并
很容易想到并查集去维护
因为花费已经排好序了,答案一定是最优的
代码
#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
struct lzf{
int c,v[5];
}a[100010];
struct node{
int to,next;
}f[1000010];
int n,t,tot,ans,fa[200010],head[200010];
void add(int x,int y)
{
f[++t].to=y;
f[t].next=head[x];
head[x]=t;
f[++t].to=x;
f[t].next=head[y];
head[y]=t;
}
bool cmp(lzf l,lzf y)
{
return l.c<y.c;
}
void dfs(int x,int ba)
{
fa[x]=ba;
for (int i=head[x];i;i=f[i].next)
if (!fa[f[i].to])
dfs(f[i].to,ba);
}
int find(int x)
{
if (x==fa[x]) return x;
else return fa[x]=find(fa[x]);
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
scanf("%d",&a[i].c);
for (int j=1;j<=4;j++)
scanf("%d",&a[i].v[j]);
add(a[i].v[1],a[i].v[2]);
add(a[i].v[3],a[i].v[4]);
}
for (int i=1;i<=2*n;i++)
if (!fa[i])
dfs(i,i);
sort(a+1,a+n+1,cmp);
for (int i=1;i<=n;i++)
{
int fx=find(a[i].v[1]),fy=find(a[i].v[3]);
if (fx!=fy)
{
fa[fx]=fy;
ans+=a[i].c;
}
}
printf("%d",ans);
return 0;
}