prim 把不符合条件的两个岛屿间的权值赋为无穷大,这样,就不会再取这两个岛屿
#include<cstdio>
#include<cstring>
#include<cmath>
#define INF 0xfffffff
double map[125][125],low[125],sum;
int x[125],y[125]; //map二维数组存图,low记录每2个点间最小权值,vis标记某点是否已访问
int vis[125];
int n;
double dlen(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 prim()
{
double min,sum=0;
int i,j,pos;
memset(vis,0,sizeof(vis));
vis[1]=1; pos=1; //从某点开始,分别标记vis和记录该点pos
for(i=1;i<=n;++i) //第一次给low数组赋值 map的第一行
low[i]=map[pos][i];
for(i=1;i<n;++i) //再运行n-1次,一次找一个最小
{
min=INF;
for(j=1;j<=n;++j)
{
if(vis[j]==0&&low[j]<min)
{
min=low[j];
pos=j;
}
}
if(min==INF) //不能连通 找不到符合条件的
{
printf("oh!\n");
return ;
}
vis[pos]=1; //标记该点已访问
sum+=min; //最小权值累加
for(j=1;j<=n;++j) //更新权值low 把 map的 pos 行中比对应的 low 小的赋给low
if(vis[j]==0&&low[j]>map[pos][j])
low[j]=map[pos][j];
}
printf("%.1lf\n",sum*100);
return ;
}
int main()
{
int T,i,j;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(i=1;i<=n;++i)
scanf("%d%d",&x[i],&y[i]);
memset(map,0,sizeof(map));
for(i=1;i<=n;++i)
{
for(j=1;j<=n;++j)
{
map[i][j]=map[j][i]=dlen(i,j);
if(map[i][j]<10||map[i][j]>1000) //不符合条件 将权值设为无穷大,不会用再到
map[i][j]=map[j][i]=INF;
}
}
prim();
}
return 0;
}
kruscal算法,其中用到了并查集的知识
下面是AC的代码
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int n;
int per[125]; // 并查集
double x[125],y[125];//小岛坐标
struct node{
int u,v;
double w; //w为距离
}s[10010];
bool cmp(node a,node b)
{
return a.w<b.w;
}
void itoa() //初始化
{
for(int i=0;i<=n;++i)
per[i]=i;
}
int find(int x) // 查找根节点
{
if(x==per[x])
return x;
return find(per[x]);
}
bool join(int a,int b) //合并根节点,并判断是否成环
{
int fa=find(a);
int fb=find(b);
if(fb!=fa)
{
per[fa]=fb;
return true;
}
return false;
}
int main()
{
int T,i,j;
scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
for(i=0;i<n;++i)
scanf("%lf%lfd",&x[i],&y[i]);
int k=0; // 两两相连 一共可建 k 个桥
double d; //两岛间距
for(i=0;i<n-1;++i)
{
for(j=i+1;j<n;++j)
{
d=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
if(d>=10&&d<=1000)
{
s[k].u=i; //小岛编号
s[k].v=j;
s[k].w=d;
k++;
}
}
}
sort(s,s+k,cmp); //按距离从小到大排序
itoa();
int flag=0;
double sum=0;
for(i=0;i<k;++i)
{
if(join(s[i].u,s[i].v)) //判断是否成环
{
sum+=s[i].w;
flag++; //已连的路数
}
}
if(flag==n-1) //m个点 m-1 条边,为一棵树
printf("%.1lf\n",sum*100);
else
printf("oh!\n");
}
return 0;
}
文章为转载,希望你能喜欢,喜欢记得关注哦!