继续畅通工程

http://acm.hdu.edu.cn/showproblem.php?pid=1879

prim:

#include <iostream>   
#include <cmath>   
  
using namespace std;  
#define MAX 99999   
#define LEN 101   
int dist[LEN];  
int map[LEN][LEN];  
bool isvisited[LEN];  
  
//初始化   
void init(){  
    int i,j;  
    for(i=0;i<LEN;i++){  
        for(j=0;j<LEN;j++){  
            if(i==j) map[i][j]=0;   //对a[][]进行初始化，一般都要；   
            map[i][j]=MAX;  
        }  
    }  
}  
  
  
//prim算法   
int prim(int n){  
    int i,j,min,pos;  
    int sum=0;  
    memset(isvisited,false,sizeof(isvisited));  
  
    //初始化   
    for(i=1;i<=n;i++){  
        dist[i]=map[1][i];  
    }  
  
    //从1开始   
    isvisited[1]=true;  
    dist[1]=MAX;  
  
    //找到权值最小点并记录下位置   
    for(i=1;i<n;i++){  
        min=MAX;  
        //pos=-1;   
        for(j=1;j<=n;j++){  
            if(!isvisited[j] && dist[j]<min){  
                min=dist[j];  
                pos=j;  
            }  
        }     
        sum+=dist[pos];//加上权值   
        isvisited[pos]=true;  
  
        //更新权值   
        for(j=1;j<=n;j++){  
            if(!isvisited[j] && dist[j]>map[pos][j]){  
                dist[j]=map[pos][j];  
            }  
        }  
    }     
    return sum;  
}  
  
int main(){  
  
    int n;  
  
    while(cin>>n){  
        init();  
        if(n==0) break;  
  
        int i,j,a,b,d,state;  
  
        for(i=0;i<n*(n-1)/2;i++){  
  
            cin>>a>>b>>d>>state;  
  
            if(state==1){  
                map[a][b]=map[b][a]=0;  
            }  
            else{  
                map[a][b]=map[b][a]=d;  
            }  
        }  
        cout<<prim(n)<<endl;  
  
    }     
     return 0;  
}  

Kruskal:

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<string.h>
struct point{
    int x,y;
    int index;
}num[5008];
int flag[102];
int cmp(const void *a,const void *b)
{
    struct point *c,*d;
    c=(struct point *)a;
    d=(struct point *)b;
    return c->index-d->index;
}
int father(int x)
{
    if(x==flag[x])
        return x;
    flag[x]=father(flag[x]);
    return flag[x];
}
int main()
{
    int n,i,sum,ra,rb,c,d;
    while(scanf("%d",&n),n)
    {
        for(i=0;i<=n;i++)
            flag[i]=i;
        for(i=0;i<n*(n-1)/2;i++)
{
scanf("%d%d%d%d",&num[i].x,&num[i].y,&c,&d);
if(d==1)num[i].index=0;
else num[i].index=c;
}
        qsort(num,n*(n-1)/2,sizeof(num[0]),cmp);
        sum=0;
        for(i=0;i<n*(n-1)/2;i++)
        {
            ra=father(num[i].x);
            rb=father(num[i].y);
            if(ra==rb)continue;
            else {
                sum+=num[i].index;
                if(ra<rb)
                    flag[rb]=ra;
                else flag[ra]=rb;
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}