1.Prim算法
时间是复杂度O(n2),适合稠密图。
例:Poj–1258
题目大意:n个城市建造光缆,要使这些城市直接通信,并且光缆费用最小。
#include <cstdio>
#include <string.h>
#define n 10010
#define inf 100010
int a[n][n],ans;
bool vis[n],t;
int dis[n];
bool prim()
{
memset(vis,0,sizeof(vis));
for(int i=1; i<=t; i++)
dis[i]=inf;
ans=0;
dis[1]=0;
for(int i=1; i<=t; i++)
{
int temp=inf,k=0;
for(int j=1; j<=t; j++)
{
if((vis[j]==0)&&(dis[j]<temp))
{
temp =dis[j];
k=j;
}
}
if(temp==n)
return false;
vis[k]=true;
ans+=temp;
for(int j=1; j<=t; j++)
{
if(vis[j]==0&&dis[j]>a[k][j])
dis[j]=a[k][j];
}
}
return true;
}
int main()
{
while(~scanf("%d",&t))
{
ans=0;
for(int i=1; i<=t; i++)
for(int j=1; j<=t; j++)
scanf("%d",&a[i][j]);
prim();
printf("%d\n",ans);
}
return 0;
}
2.Kruskal算法
时间复杂度O(elog2e),适合简单图。
例如:Poj–2485
题目大意:给定一些地点和连通它们的可以修的公路,每条路有不同长度,求修建一些路使它们连通,并且所有路长度最小。
#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 1001
#define inf 1000000001
#define M N*N
int t;
int e,n,m;
int f[N];
struct note
{
int u,v;
int c;
note() {}
note(int u,int v,int c):u(u),v(v),c(c) {}
} p[M];
void add(int u,int v,int c)
{
p[e++]=note (u,v,c);
}
int cmp(const void * a,const void * b)
{
note *aa=(note *)a;
note *bb=(note *)b;
return aa->c-bb->c;
}
void made()
{
int i;
for(i=0; i<=n; i++)
f[i]=i;
}
int find(int x)
{
if(x!=f[x])
f[x]=find(f[x]);
return f[x];
}
int kru(int n)
{
qsort(p,e,sizeof(p[0]),cmp);
made();
int m=0,ans=0;
int i,j;
for(i=0; i<e; i++)
{
int xx=find(p[i].u);
int yy=find(p[i].v);
if(xx==yy)
continue;
m++;
ans =p[i].c;
f[yy]=xx;
if(m==n-1)
break;
}
if(m<n-1)
return -1;
else return ans;
}
int main()
{
int i,j,l,c;;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
e=0;
for(i=1; i<=n; i++)
for(j=1; j<=n; j++)
{
scanf("%d",&c);
add(i,j,c);
}
printf("%d\n",kru(n));
}
return 0;
}