hdu 1162

主题思想，最小生成树，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;
}