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