HDU1879-继续畅通工程（Prim算法+Kruskal算法）

来源：互联网发布：app软件怎么制作编辑：程序博客网时间：2024/05/23 12:11

继续畅通工程

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 22240 Accepted Submission(s): 9500

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

当N为0时输入结束。

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

Sample Input

31 2 1 01 3 2 02 3 4 031 2 1 01 3 2 02 3 4 131 2 1 01 3 2 12 3 4 10

Sample Output

Author

ZJU

Source

浙大计算机研究生复试上机考试-2008年

Prim算法：

#include <iostream>#include <string.h>#include <stdio.h>using namespace std;int map[105][105];int dis[105];int visit[105];const int INF=0x3f3f3f3f;int main(){    int n;    while(~scanf("%d",&n)&&n)    {        int N=n*(n-1)/2;        memset(map,0,sizeof map);        memset(visit,0,sizeof visit);        int a,b,c,d;        for(int i=0;i<N;i++)        {            scanf("%d %d %d %d",&a,&b,&c,&d);            if(d==1) c=0;            map[a][b]=map[b][a]=c;        }        visit[1]=1;        for(int i=1;i<=n;i++)            dis[i]=map[1][i];        int sum=0;        for(int i=1;i<n;i++)        {            int mi=INF;            int k=-1;            for(int j=1;j<=n;j++)                if(!visit[j]&&dis[j]<mi)                mi=dis[j],k=j;            if(k!=-1)            {                sum+=mi;                visit[k]=1;                for(int j=1;j<=n;j++)                {                    if(!visit[j]&&dis[j]>map[k][j])                        dis[j]=map[k][j];                }            }        }        printf("%d\n",sum);    }    return 0;}

Kruskal算法：

#include <iostream>#include <string.h>#include <stdio.h>#include <math.h>#include <algorithm>using namespace std;struct Edge{    int a,b,c;}x[10005];int n,N;int parent[105];int cmp(Edge p,Edge q){   return p.c<q.c;}int Find(int x){    int s=x;    while(parent[s]>=0) s=parent[s];    while(x!=s)    {        int tmp=parent[x];        parent[x]=s;        x=tmp;    }    return s;}void kruscal(){    int sum=0;    int num=0;    memset(parent,-1,sizeof parent);    for(int i=1;i<=N;i++)    {        int a=x[i].a;int b=x[i].b;        if(Find(a)!=Find(b))        {            sum+=x[i].c;num++;            int a1=Find(a),b1=Find(b);            parent[a1]=b1;        }        if(num>=n-1) break;    }    printf("%d\n",sum);}int main(){    while(~scanf("%d",&n)&&n)    {        int a,b,c,d;        N=(n-1)*n/2;        for(int i=1;i<=N;i++)        {            scanf("%d %d %d %d",&a,&b,&c,&d);            if(d==1) c=0;            x[i].a=a,x[i].b=b,x[i].c=c;        }        sort(x+1,x+1+N,cmp);        kruscal();    }    return 0;}

0 0