Prim最小生成树【poj 2349 Arctic Network;poj 1287 NetWorking】

poj 2349 Arctic Network:

图我们构造完成，调用prim算法，把该算法中每一次贪心得到的最小权边存入一个数组中。算法结束后对那个数组sort递减排序。这样，前边最大的s个用那卫星让他们通信去。然后从第s开始配给他们足够功率的对讲机使。

#include <iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAX 1003
#define VALUE 999999999
using namespace std;

struct distinct
{
int x;
int y;
};

double g[MAX][MAX];
double minCost[MAX];
int visited[MAX];
double arr[MAX];
//求两点之间的距离
double dis(int x1,int y1,int x2,int y2)
{
return sqrt((x2-x1)*(x2-x1)*1.0+(y2-y1)*(y2-y1)*1.0);
}

//求最小生成树
void prim(int n)
{
int i;
int temp=0;
for(i=0;i<n;i++)
{
visited[i]=0;
minCost[i]=g[0][i];
arr[i]=0;
}
minCost[0]=0;
while(true)
{
int t=-1;
for(i=0;i<n;i++)
{
if(visited[i]==0 && (t==-1 || minCost[i]<minCost[t]))
{
t=i;
}
}
if(t==-1)
break;
visited[t]=1;
arr[temp]=minCost[t];//保存每一条最短路径
temp++;
//printf("%d\t",minCost[t]);
for(i=0;i<n;i++)
{
if(minCost[i]>g[i][t] && g[i][t]!=0)
{
minCost[i]=g[i][t];
}
}
}
}

bool cmp(double i,double j) //比较函数。
{
return i>j;
}

int main()
{
int ts;
int i,j;
scanf("%d",&ts);
while(ts--)
{
memset(g,0,sizeof(g));
int n,m;
distinct d[MAX];
memset(d,0,sizeof(d));
scanf("%d%d",&n,&m);
for(i=0;i<m;i++)
{
scanf("%d%d",&d[i].x,&d[i].y);
for(j=0;j<i;j++)
{//建立无向完全图
g[i][j]=dis(d[i].x,d[i].y,d[j].x,d[j].y);
g[j][i]=g[i][j];
}
}
prim(m);

//对数组minCost进行从大到小排序
sort(arr,arr+m,cmp);

if(n==0)
{
printf("%.2lf\n",arr[0]);
}
else
printf("%.2lf\n",arr[n-1]);
}
return 0;
} poj 1287 NetWorking

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
#define MAXN 60
#define INF 0xFFFFFFF

int n , m;
int vis[MAXN];
int lowcost[MAXN];
int G[MAXN][MAXN];

void init(){
memset(vis , 0 , sizeof(vis));
memset(lowcost , 0 , sizeof(lowcost));
for(int i = 1 ; i <= n ; i++){
for(int j = 1 ; j <= n ; j++)
G[i][j] = INF;
}
}

void prime(){
int ans , pos;
for(int i = 1 ; i <= n ; i++)
lowcost[i] = G[1][i];
ans = 0;
vis[1] = 1;
for(int i = 1 ; i <= n ; i++){
pos = -1;
for(int j = 1 ; j <= n ; j++){
if(!vis[j] && (pos == -1 || lowcost[j] < lowcost[pos]))
pos = j;
}

if(pos == -1)
break;

vis[pos] = 1;
ans += lowcost[pos];
for(int j = 1 ; j <= n ; j++){
if(!vis[j] && lowcost[j] > G[pos][j])
lowcost[j] = G[j][pos];
}
}
printf("%d\n" , ans);
}

int main(){
int x , y , v;
while(scanf("%d%d" , &n , &m) && n){
init();
if(!m)
printf("0\n");
else{
for(int i = 0 ; i < m ; i++){
scanf("%d%d%d" , &x , &y , &v);
if(G[x][y] > v)
G[x][y] = G[y][x] = v;
}
prime();
}
}
system("pause");
return 0;
}