kruskal算法
建图之后,首先对边的权值排序,每次取出没有被选中并且最小的边,用并查集判断边的两点是否相连,如果相连则跳过找下一个点,否则连接着两个点,当连接的边数等于点数-1时,就找到了最小生成树。
例题:Networking POJ - 1287
#include<iostream>
#include<algorithm>
#include<cstdio>
#define max_n 55
using namespace std;
int father[max_n];
struct edge{
int from,to;
long long cost;
}e[max_n*max_n];
bool cmp(edge a,edge b)
{
return a.cost<b.cost;//自定义cmp
}
int findd(int x)
{
if(x==father[x])return x;
else
return father[x]=findd(father[x]);//路径压缩
}
int main()
{
int p,r;
while(scanf("%d%d",&p,&r)!=EOF,p)
{
long long ans=0;
for(int i=1;i<=p;i++)
{
father[i]=i;//初始化
}
for(int i=1;i<=r;i++)
{
cin>>e[i].from>>e[i].to>>e[i].cost;
}
sort(e+1,e+1+r,cmp);//对边的权值排序
for(int i=1;i<=r;i++)
{
int a=findd(e[i].from);//判断两个点是否已经相连
int b=findd(e[i].to);
if(a!=b)//如果没有相连则连起来,ans加上这条边的权值
{
ans+=e[i].cost;
father[a]=b;
}
}
cout<<ans<<endl;
}
return 0;
}
prim
核心思路是贪心,每次找当前没有选中的最小边连接起来,可以用priority_queue每次弹出当前最小的边。
例题:还是畅通工程 HDU - 1233
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
typedef long long ll;
int n;
const int maxx =110 ;
const int inf=0x3f3f3f3f;
int mapp[maxx][maxx];//存图,为了省空间也可以用邻接表,
bool visited[maxx];
struct edge{
int u, v, w;
edge(int u, int v, int w) {
this->u = u;
this->w = w;
this->v = v;
}
edge() {};
bool operator<(const edge& a)const {//重载小于号,可以建立小根堆
return a.w < w;
}
}e[maxx*maxx];
std::priority_queue<edge>q;
ll prim() {
memset(visited,0,sizeof(visited));
int cnt=0;
int ans = 0;
for (int i = 1; i <= n; i++) {
if(mapp[1][i]!=inf){
q.push(edge(1,i,mapp[1][i]));//首先把跟点1相连的边加入优先队列
}
}
visited[1]=1;
while(!q.empty()){
edge temp=q.top();//每次弹出当前队列中权值最小的边
q.pop();
if(visited[temp.v])continue;//如果这条边的到达点已经被标记过就continue
visited[temp.v]=1;
ans+=temp.w;
cnt++;
if(cnt==n-1)return ans;//如果加入的边等于点数-1,说明已经找到了最小生成树
for(int i=1;i<=n;i++){
if(mapp[temp.v][i]!=inf)//把跟这个点连接的边全部加入队列
q.push(edge(temp.v,i,mapp[temp.v][i]));
}
}
return ans;
}
int main() {
while (scanf("%d", &n) != EOF) {
if(n==0)break;
memset(mapp,inf,sizeof(mapp));
for(int i=1;i<=n;i++){
mapp[i][i]=0;//初始化
}
while (!q.empty())q.pop();
for (int i = 1; i <= n * (n - 1) / 2; i++) {
int x, y, z;
scanf("%d%d%d", &x, &y, &z);
mapp[x][y] = mapp[y][x] = z;
}
ll ans = prim();
for (int i = 1; i <= n; i++) {
if (!visited[i])ans = -1;
}
if (ans == -1)printf("?\n");
else
printf("%lld\n", ans);
}
return 0;
}