prim算法与kruscal算法（例题 :HDU 1875）解决最小生成树

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;
}

文章为转载，希望你能喜欢，喜欢记得关注哦！