zoj-1203(Prim算法)

MST是一個非常非常經典的問題了,ZOJ-1203則是一題MST的裸題,題目的意思就是給定n個座標,連起n個點最短是多長。

我用的是prim算法,創建圖什麼的這個大家都懂。

(好久沒寫過題解了,不知道說些什麼好~哎。)


/***********************************************************
	> OS     : Linux 3.11.10-301.fc20.x86_64 Fedora20
	> Author : yaolong
	> Mail   : [email protected] 
	> Time   : 2014年07月27日 星期日 13時23分30秒
 **********************************************************/
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<cmath>
using namespace std;
#define INF 1<<30 
class Point{
    public:
        Point(){}
        double x,y;
    double dist(Point &pt){        
        return sqrt((x-pt.x)*(x-pt.x)+(y-pt.y)*(y-pt.y));
    }
};
Point v[105];
double mp[105][105];
double low[105];
bool used[105];
double prim(int n){
    double result=0;
    double min;
    int cnt=n-1;
    int pos,i;
    memset(used,0,sizeof(used));
    //原始點
    used[0]=1;
    for(i=0;i<n;i++){
        low[i]=mp[0][i];
    }

    while(cnt--){
        min=INF;
        for(i=0;i<n;i++){
            if(!used[i]&&low[i]<min){
                pos=i;
                min=low[i];
            }
        }
        result+=min;
        used[pos]=1;
        for(i=0;i<n;i++){
            if(low[i]>mp[pos][i]){
                low[i]=mp[pos][i];
            }
        }
    }
    return result;
}
int main(){
    int n,i,j;
    double x,y;
    int cas=1;
    bool flag=0;
    while(scanf("%d",&n),n){
        if(flag){
            puts("");        
        }else{            
            flag=1;        
        }
        for(i=0;i<n;i++){
            scanf("%lf%lf",&v[i].x,&v[i].y);
                    
        }
        for(i=0;i<n-1;i++){
            for(j=i+1;j<n;j++){
                mp[i][j]=mp[j][i]=v[i].dist(v[j]);
            }
        }
        printf("Case #%d:\n",cas);
        cas++;
        printf("The minimal distance is: %.2lf\n",prim(n));
    }

   
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章