Prim
本质是维护一个集合,选择一个点作为初始点A先加入集合,遍历未加入集合的点,将它们到A的距离设为到集合的最短距离,选则其中的最小值,将该点再加入集合。每次有新的点加入集合后,都用该点更新其余未加入集合的点到集合的距离。(与dijkstra不同的是,prim每次选择到集合最近的点,而dijkstra选择到源点最近的点)。
const int INF=1e9;
const int maxn=2e3+5;
int cost[maxn][maxn];
int lowc[maxn],vis[maxn];
ll prim(){
ll ans=0;
memset(vis,0,sizeof(vis));
vis[1]=1;
for(int i=2;i<=n;i++) lowc[i]=cost[1][i];
for(int i=2;i<=n;i++){
int minn=INF,p=-1;
for(int j=1;j<=n;j++){
if(!vis[j]&&lowc[j]<minn){
minn=lowc[j];
p=j;
}
}
if(minn==INF)return -1;
ans+=minn;
vis[p]=1;
for(int j=1;j<=n;j++){
if(!vis[j]&&lowc[j]>cost[p][j])
lowc[j]=cost[p][j];
}
}
return ans;
}
kruskal
建立边结构体,将边按照权值排序,从小到大加入树中,若一条边的两个点已经在同一棵树中(用并查集维护),则跳过这条边。
struct edge{
int f,t,w;
}E[maxm];
int fa[maxn];
int find(int x){
return x==fa[x]?x:fa[x]=find(fa[x]);
}
ll kruskal(){
ll ans=0;
sort(E+1,E+1+tot,cmp);
for(int i=1;i<=n;i++)
fa[i]=i;
int cnt=0;
for(int i=1;i<=tot;i++){
int u=E[i].f;
int v=E[i].t;
u=find(u);v=find(v);//跳到祖先节点
if(u!=v){
ans+=E[i].w;
fa[u]=v;
cnt++;
}
if(cnt==n-1)return ans;
}
if(cnt==n-1)return ans;
else return -1;
}