最小生成树-Prime
文章目录
Prime
Prime算法,其实就是加点。Prime算法先将起点加入到最小生成树上,之后不断寻找与最小生成树相连的边权最小所连接的点,并将该点加入到最小生成树中。最小生成树总共有n-1条边。因为是加点,因此Prime算法适用于点少而边多的图。
算法思想
取图中任意一个顶点v作为生成树的根,之后往生成树上添加新的顶点 w。在添加的顶点w和已经在生成树上的顶点v之间必定存在一条边,并且该边的权值在所有连通顶点v和w之间的边中取值最小。之后继续往生成树上添加顶点,直至生成树上含有n-1条边为止。
模板
int prime(){
memset(dist,INF,sizeof(dist));
memset(vis,0,sizeof(vis));
int ans=0;
for(int i=0;i<n;i++){
int t=-1;
for(int j=1;j<=n;j++){
if(!vis[j] && (t==-1 || dist[j]<dist[t])){
t=j;
}
}
vis[t]=1;
if(i) ans+=dist[t];
for(int j=1;j<=n;j++){
if(vis[j]) continue;
dist[j]=min(dist[j],a[t][j]);
}
}
return ans;
}
例题
POJ-1287 Networking
题目描述
存在许多点和点与点之间的路径,路径长度不一,点到点之间可能存在多条路径。挑选部分路径使得所有点连通且总路径长度最小。
Input
多样例输入。每个样例都保证有解。
每个样例的第一行有两个整数,P(点的个数),R(点与点之间的路径总数)。
接下来的R行输入路径,每行含3个整数,前两个数表示连接的两个点,第三个数表示路径长度。
当P为0时输入结束。样例与样例之间存在空行
P最大为50,路径长度最大为100,路径数没有限制。
i和j之间的路径可以表示为 i j 或 j iOutput
对于每个样例,输出一个数表示设计出的路径总长度
Sample Input
1 0 2 3 1 2 37 2 1 17 1 2 68 3 7 1 2 19 2 3 11 3 1 7 1 3 5 2 3 89 3 1 91 1 2 32 5 7 1 2 5 2 3 7 2 4 8 4 5 11 3 5 10 1 5 6 4 2 12 0
Sample Output
0 17 16 26
思路分析
最小生成树模板题目,此题用Prime算法。
详细代码
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int INF=0x3f3f3f3f;
const int maxn=52;
int n,r;
int a[maxn][maxn],dist[maxn];
bool vis[maxn];
int prime(){
memset(dist,INF,sizeof(dist));
memset(vis,0,sizeof(vis));
int ans=0;
for(int i=0;i<n;i++){
int t=-1;
for(int j=1;j<=n;j++){
if(!vis[j] && (t==-1 || dist[j]<dist[t])){
t=j;
}
}
vis[t]=1;
if(i) ans+=dist[t];
for(int j=1;j<=n;j++){
if(vis[j]) continue;
dist[j]=min(dist[j],a[t][j]);
}
}
return ans;
}
int main(){
while(scanf("%d",&n) && n){
memset(a,INF,sizeof(a));
scanf("%d",&r);
for(int i=0;i<r;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
a[u][v]=a[v][u]=min(w,a[u][v]);
}
printf("%d\n",prime());
}
return 0;
}
POJ-2031 Building a Space Station
题目描述
rbbb有N个魔法球,每个魔法球的中心位于三维空间的(x,y,z)坐标,假定每个魔法球的魔法力场为以其中心为圆心的半径为r的球体,若两个球体之间相交相切或相包含,则两个魔法球可以直接相连,否则需要在两个魔法球之间需要通过魔力连接,连接两个魔法球消耗的法力值为两个魔法球中心相连的距离减去两个半径,现在需要使任意两个魔法球都能直接或间接相连,求所需的最少法力值。 N<=100,坐标非负且小于100.000。
Input
多组输入,第一行为N,接下来N行每行四个数字,分别表示第i个魔法球中心坐标x,y,z和魔法力场范围半径r,以N=0结束。
Output
所需的最少法力值
Sample Input
5 10.000 10.000 10.000 10.000 20.000 20.000 20.000 20.000 30.000 30.000 30.000 30.000 40.000 40.000 40.000 40.000 50.000 50.000 50.000 50.000 3 12.345 23.456 34.567 1.234 1.234 2.345 3.456 2.345 4.321 5.432 6.543 3.456 0
Sample Output
0.000 29.582
思路分析
模板题
详细代码
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const double INF=102;
const int maxn=102;
int n;
double a[maxn][maxn],dist[maxn];
bool vis[maxn];
struct node{
double x,y,z,r;
}m[maxn];
double getDistance(node a,node b){
return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y)+(a.z-b.z)*(a.z-b.z));
}
double prime(){
for(int i=1;i<=n;i++) dist[i]=INF;
memset(vis,0,sizeof(vis));
double ans=0;
for(int i=0;i<n;i++){
int t=-1;
for(int j=1;j<=n;j++){
if(!vis[j] && (t==-1 || dist[j]<dist[t])){
t=j;
}
}
vis[t]=1;
if(i) ans+=dist[t];
for(int j=1;j<=n;j++){
if(vis[j]) continue;
dist[j]=min(dist[j],a[t][j]);
}
}
return ans;
}
int main(){
while(scanf("%d",&n) && n){
for(int i=1;i<=n;i++){
cin>>m[i].x>>m[i].y>>m[i].z>>m[i].r;
}
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
a[i][j]=getDistance(m[i],m[j])-m[i].r-m[j].r;
if(a[i][j]<=0) a[i][j]=0;
}
}
printf("%.3lf\n",prime());
}
return 0;
}