hdu 1162

最新推荐文章于 2021-03-18 07:09:37 发布

原创最新推荐文章于 2021-03-18 07:09:37 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

hdu 同时被 2 个专栏收录

119 篇文章

订阅专栏

prim

4 篇文章

订阅专栏

主题思想，最小生成树，prim算法

prim算法最好的情况下使用优先队列，所有边的集合E每次输出距离最小生成树节点距离最小的边。到MST 集合中，MST代表组成最小生成树的集合。

这里利用循环来代替优先队列。核心思想还是每次选取最小的。
最小生成树代码

double  prim(){
    vector<int> v;
    double ans=0;
    memset(visited,false,sizeof(visited));
    v.push_back(0);
    visited[0]=true;
    while(true){
        if(v.size()==n)break;
        int len=v.size();
        double mm=INF;
        int pre=0;
        int index=-1;
        //mst  set
        for(int i=0;i<len;i++){
            //all edge set
            for(int j=0;j<n;j++){
                // e-mst set
                if(j!=v[i]&&!visited[j]){
                    if(dist[v[i]][j]<mm){
                        mm=dist[v[i]][j];
                        pre=v[i];
                        index=j;
                    }
                }
            }
        }
        v.push_back(index);
        visited[index]=true;
        ans+=dist[pre][index];
    }
    return ans;

}

AC 代码

#include <iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
const int maxn=105;

const double eps=1e-8;

const double INF=1<<30-1;
using namespace std;

struct Point{

    double x;
    double y;

};
Point point[maxn];
double Distance(Point a,Point b){
    double dist;

    dist=sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
    return dist;
}
int n;

double dist[maxn][maxn];
bool visited[maxn];


double  prim(){
    vector<int> v;

    double ans=0;
    memset(visited,false,sizeof(visited));

    v.push_back(0);
    visited[0]=true;
    while(true){

        if(v.size()==n)break;

        int len=v.size();

        double mm=INF;
        int pre=0;
        int index=-1;
        for(int i=0;i<len;i++){
            for(int j=0;j<n;j++){

                if(j!=v[i]&&!visited[j]){

                    if(dist[v[i]][j]<mm){

                        mm=dist[v[i]][j];
                        pre=v[i];
                        index=j;

                    }
                }
            }
        }
        v.push_back(index);
        visited[index]=true;
        ans+=dist[pre][index];
    }

    return ans;

}
int main()
{

    while(scanf("%d",&n)!=EOF){

        for(int i=0;i<n;i++){

            scanf("%lf%lf",&point[i].x,&point[i].y);

            dist[i][i]=0;
        }

        for(int i=0;i<n;i++){

            for(int j=i+1;j<n;j++){
                dist[i][j]=Distance(point[i],point[j]);
                dist[j][i]=dist[i][j];
            }
        }

        //prim

        double ans=prim();
        printf("%.2lf\n",ans);
    }

    return 0;
}