codevs上的一道模板题,最小生成树的意思大概就是,给好多条边和权值,让用最小的代价(权值和)使这个图连通
如果这个图一共有n个点,那么做法就是对于每一个边的权值给边排序,从小到大,然后从小里选,如果选出的边的两个点已经在一个集合里了
就跳过,如果不在一个集合里,就并上这两点,同时ans加上这条边的权值,由于排过序,所以每次并边的代价一定是最小的,并且记录并边的
次数,如果次数==n-1,说明这个图就连通了,退出即可
#include<iostream>
#include<algorithm>
using namespace std;
struct line
{
int from;
int to;
int w;
}edge[100001];
int map[101][101];
int n;
int tot;
int pre[101];
int rank[101];
void init()
{
for(int i=1;i<=n;i++)
{
pre[i]=i;
rank[i]=0;
}
}
int find(int x)
{
if(x==pre[x])return x;
return pre[x]=find(pre[x]);
}
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(rank[x]>rank[y])pre[y]=x;
else
{
pre[x]=y;
if(rank[x]==rank[y])rank[y]++;
}
}
bool cmp(const line& a,const line& b)
{
return a.w<b.w;
}
int main()
{
std::ios::sync_with_stdio(false);
cin>>n;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
cin>>map[i][j];
if(map[i][j]==0)continue;
edge[++tot].from=i;
edge[tot].to=j;
edge[tot].w=map[i][j];
}
init();
int ans=0;
int k=0;
sort(edge+1,edge+1+tot,cmp);
for(int i=1;i<=tot;i++)
{
if(find(edge[i].from)!=find(edge[i].to))
{
unite(edge[i].from,edge[i].to);
ans+=edge[i].w;
k++;
}
if(k==n-1)break;
}
cout<<ans;
return 0;
}