Kruscal
第一种:
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
struct Island
{
int x,y;
};
struct node
{
int u,v;
double w;
};
Island arr[220];
node edge[20000];
int per[220];
int n;
bool cmp(node a,node b)
{
return a.w<b.w;
}
void init()
{
for(int i=1;i<=n;++i)
{
per[i]=i;
}
}
int find(int x)
{
if(x==per[x]) return x;
return per[x]=find(per[x]);
}
bool join(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)
{
per[fx]=fy;
return 1;
}
return 0;
}
int main()
{
int T;
int i,j,k;
double x,y;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
init();
for(i=1;i<=n;++i)
{
scanf("%d%d",&arr[i].x,&arr[i].y);
}
//m=n*(n-1)/2;
k=0;
for(i=1;i<=n;++i)//把所有路记录在node结构体中
{
for(j=i+1;j<=n;++j)
{
edge[k].u=i;
edge[k].v=j;
x=(arr[j].y-arr[i].y)*(arr[j].y-arr[i].y);
y=(arr[j].x-arr[i].x)*(arr[j].x-arr[i].x);
double temp=sqrt(x+y);
edge[k++].w=temp;
}
}
sort(edge,edge+k,cmp);
double sum=0;
for(i=0;i<k;++i)
{
if(edge[i].w<=1000&&edge[i].w>=10&&join(edge[i].u,edge[i].v))//如果两个岛的距离不符合要求就会把join(edge[i].u,edge[i].v)短路
{
sum+=edge[i].w;
}
}
int cnt=0;
bool flag=0;
for(i=1;i<=n;++i)//短路了就不会执行了,也就不会连接了,就只需要判断根节点的个数
{
if(i==per[i]) cnt++;
if(cnt>1) //不等于1就还有元素(小岛)没连起来,不满足题意
{
flag=1;
break;
}
}
if(flag) printf("oh!\n");
else
printf("%.1lf\n",sum*100);
}
return 0;
}
第二种:
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
int k,c,u[10005],v[10005],sum,r[10005],p[200],a[200][2];
double w[10005],cnt;
int dd(int a,int b,int c,int d);
int kruskal();
int main()
{
int t;
scanf("%d",&t);
while(t--)
{k=0;
scanf("%d",&c);
for(int i=0;i<c;i++)
{
scanf("%d%d",&a[i][0],&a[i][1]);
for(int j=0;j<i;j++)
{
if(dd(a[i][0],a[i][1],a[j][0],a[j][1])<=1000000&&dd(a[i][0],a[i][1],a[j][0],a[j][1])>=100)
{
u[k]=i;
v[k]=j;
w[k]=sqrt((double)(dd(a[i][0],a[i][1],a[j][0],a[j][1])));
//cout<<i<<"***"<<j<<"***"<<k<<"***"<<w[k]<<"\n\n\n";
//cout<<dd(1,1,2,2)<<"*&*&";
k++;
}
}
}kruskal();
}return 0;
}
int dd(int a,int b,int c,int d){return ((a-c)*(a-c)+(b-d)*(b-d));}
int cmp(int i,int j){return w[i]<w[j];}
int find(int x){return x==p[x]?x:(p[x]=find(p[x])); }
int kruskal()
{sum=0;cnt=0;
//cout<<k<<"***\n";
for(int i=0;i<c;i++)p[i]=i;
for(int i=0;i<k;i++)r[i]=i;
sort(r,r+k,cmp);
for(int i=0;i<k;i++)
{
int e=r[i];
int x=find(u[e]),y=find(v[e]);
if(x!=y){sum++;cnt+=w[e];p[x]=y;}
}
if(sum==c-1)printf("%.1lf\n",cnt*100.0);
else printf("oh!\n");
return 0;
}
Prim:
#include <stdio.h>
#include <string.h>
#include <math.h>
#define N 110
#define INF 0x3f3f3f3f
int n;
int i,j;
double map[N][N];
bool vis[N];//标记是否已经放入最小生成树的那个集合里了
double low[N];//记录不在已经加入最小生成树的这个集合里的元素到这个 集合的最小距离
int x[N],y[N];
double dis(int i,int j)
{
return sqrt(1.0*(x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
}
void input()
{
for(i=1;i<=n;++i)
scanf("%d%d",&x[i],&y[i]);
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
{
map[i][j]=dis(i,j);
if(map[i][j]>1000||map[i][j]<10)
{
map[i][j]=INF;
}
}
}
void prim()
{
double sum=0;
memset(vis,0,sizeof(vis));
int pos=1;//从1开始
for(i=1;i<=n;++i)//第一次给low赋值
{
low[i]=map[1][i];
}
vis[1]=1;
//已经找到一个点1,再找n-1个
for(i=1;i<n;++i)
{
double min=INF;
for(j=1;j<=n;++j)
{
if(!vis[j]&&min>low[j])//找下一个点到这个集合的最小值
{
min=low[j];//记下这个最小值
pos=j;//记下这个点
}
}
if(min==INF)
{
printf("oh!\n");
return ;
}
sum+=min;
vis[pos]=1;//把刚刚找到的这个点加入集合
for(j=1;j<=n;++j) //更新low数组
{
if(!vis[j]&&low[j]>map[pos][j])
{
low[j]=map[pos][j];
}
}
}
printf("%.1lf\n",sum*100);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
input();
prim();
}
return 0;
}