prim算法:
类似于dij算法,将每个点距离已经加入的点的集合的最短距离算出,每次取出这些距离里最短的点加入集合,并在每次加入新的点后更新这些距离即可。
算法模版:
#define INF 32767;
void Prim(MGraph g,int v)
{
int lowcost[MAXV];
int min;
//就是对应的元素下标的顶点的连接的上一个顶点
int closest[MAXV];
int i,k;
for(i=0;i<g.n;i++)
{
lowcost[i] = g.edges[v][i];
closest[i] = v;
}
for(i=1;i<g.n;i++)
{
min = INF;
//先遍历出lowcost里面最小的拿出来
for(j=0;j<g.n;j++)
{
if(lowcost[j]!=0 && lowcost[j]<min)
{
min = lowcost[j];
k = j;
}
}
printf("边(%d,%d)权为:%d\n",closest[k],k,min);
lowcost[k] = 0;
//接下来做的就是把集合里的俩个顶点跟集合外的顶点组成的边的权值做比较
for(j=0;j<g.n;j++)
{
if(g.edges[k][j]!=0 && lowcost[j]>g.edges[k][j])
{
lowcost[j] = g.edges[k][j];
closest[j] = k;
}
}
}
}
krustral算法
先将所有的边按照边权值排序,之后利用并查集,检查边的两个点是否被连通,若被连通则忽略,否则加入树中。
#include"iostream"
#include"cstdio"
#include"cstring"
using namespace std;
typedef struct node
{
int x,y;
int weight;
}e;
e edge[100];
int fa[100];
int deep[100];
int n,m;
int sum=0;
void init()
{
for(int i=1;i<=n;i++)
{
fa[i]=i;
deep[i]=1;
}
}
int find(int x)
{
if(x==fa[x]) return x;
else return fa[x]=find(fa[x]);
}
void unit(int x,int y)
{
x=find(x);
y=find(y);
if(x==y) return ;
else
{
if(deep[x]>deep[y]) fa[y]=x;
else
{
fa[x]=y;
if(deep[x]==deep[y]) deep[y]++;
}
}
}
bool same(int x,int y)
{
return find(x)==find(y);
}
void quicksort(int left,int right)
{
if(left>right) return ;
else
{
int i=left;
int j=right;
e t;
e temp=edge[left];
while(i!=j)
{
while(i<j&&edge[j].weight>=temp.weight) j--;
while(i<j&&edge[i].weight<=temp.weight) i++;
if(i<j)
{
t=edge[i];
edge[i]=edge[j];
edge[j]=t;
}
}
edge[left]=edge[i];
edge[i]=temp;
quicksort(left,i-1);
quicksort(i+1,right);
return ;
}
}
int main()
{
cin>>n>>m;
for(int i=1;i<=m;i++) cin>>edge[i].x>>edge[i].y>>edge[i].weight;
init();
quicksort(1,m);
for(int i=1;i<=n-1;)
{
if(!same(edge[i].x,edge[i].y))
{
unit(edge[i].x,edge[i].y);
sum+=edge[i].weight;
i++;
}
}
cout<<sum<<endl;
return 0;
}
次小生成树:
下面介绍一下利用prim求次小生成树的主要步骤。
1.先求出来最小生成树。并将最小生成树任意两点之间路径当中的权值最大的那一条找出来,为什么要找最大的呢,因为生成树加入一条边之后一定构成了回路,那么肯定要去掉这个回路当中一条边才是生成树,那么,怎么去边才是次小的,那就去掉除了刚刚添加的一条边之外回路当中权值最大的一个,所以留下的就是最小的。
2.枚举最小生成树外的每一条边。找出最小的就是次小生成树。
POJ1679 The Unique MST
题意:给定图,让求它的最小生成树是否唯一。如果唯一的话输出最小生成树的权值和,否则输出Not Unique!
思路:直接求次小生成树就行。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 111;
const int inf = 0x3f3f3f3f;
int Map[maxn][maxn];//邻接矩阵存图
int Max[maxn][maxn];//表示最小生成树中i到j的最大边权
bool used[maxn][maxn];//判断该边是否加入最小生成树
int pre[maxn];
int dis[maxn];
void init(int n)
{
for (int i = 1; i <= n; i++)
for (int j = 1; j <= n; j++)
if (i == j) Map[i][j] = 0;
else Map[i][j] = inf;
}
void read(int m)
{
int u, v, w;
for (int i = 0; i < m; i++)
{
scanf("%d %d %d", &u, &v, &w);
Map[u][v] = Map[v][u] = w;
}
}
int prim(int n)
{
int ans = 0;
bool vis[maxn];
memset(vis, false, sizeof(vis));
memset(used, false, sizeof(used));
memset(Max, 0, sizeof(Max));
for (int i = 2; i <= n; i++)
{
dis[i] = Map[1][i];
pre[i] = 1;
}
pre[1] = 0;
dis[1] = 0;
vis[1] = true;
for (int i = 2; i <= n; i++)
{
int min_dis = inf, k;
for (int j = 1; j <= n; j++)
{
if (!vis[j] && min_dis > dis[j])
{
min_dis = dis[j];
k = j;
}
}
if (min_dis == inf) return -1;//如果不存在最小生成树
ans += min_dis;
vis[k] = true;
used[k][pre[k]] = used[pre[k]][k] = true;
for (int j = 1; j <= n; j++)
{
if (vis[j]) Max[j][k] = Max[k][j] = max(Max[j][pre[k]], dis[k]);
if (!vis[j] && dis[j] > Map[k][j])
{
dis[j] = Map[k][j];
pre[j] = k;
}
}
}
return ans;//最小生成树的权值之和
}
int smst(int n, int min_ans)//min_ans 是最小生成树的权值和
{
int ans = inf;
for (int i = 1; i <= n; i++)//枚举最小生成树之外的边
for (int j = i + 1; j <= n; j++)
if (Map[i][j] != inf && !used[i][j])
ans = min(ans, min_ans + Map[i][j] - Max[i][j]);
if (ans == inf) return -1;
return ans;
}
void solve(int n)
{
int ans = prim(n);
if (ans == -1)
{
puts("Not Unique!");
return;
}
if (smst(n, ans) == ans)
printf("Not Unique!\n");
else
printf("%d\n", ans);
}
int main()
{
int T, n, m;
scanf("%d", &T);
while (T--)
{
scanf("%d %d", &n, &m);
init(n);
read(m);
solve(n);
}
return 0;
}