ACM第四次练习—1005

题意：省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

思路：这个题和1004差不多，就是加了一个状态，可以先判断每条路的状态，如果是修好的，就让它们的成本是0；如果没修，就让成本给出的成本。这样问题的处理就和1004一样了。

感想：虽然是个与以前做过的题类似的题，但还是废了很大劲~

代码：

#include<stdio.h>  

#include<string.h>  
#include<algorithm>  
using namespace std;  
#define inf 0x3f3f3f3f  
int map[100][100];  
int s[100],vis[100];  
int n,m;  
int prim()  
{  
    int i,j,t,p,min,cnt,minpos;  
    int ans=0;  
    cnt=0;  
    vis[1]=1;  
    s[cnt++]=1;  
    while(cnt<n)  
    {  
        t=cnt;  
        min=inf;  
        for(i=0;i<t;i++)  
        {  
            p=s[i];  
            for(j=1;j<=n;j++)  
            {  
                if(!vis[j]&&map[p][j]<min)  
                {  
                    min=map[p][j];  
                    minpos=j;  
                }  
            }  
        }  
        ans+=min;  
        s[cnt++]=minpos;  
        vis[minpos]=1;  
    }  
    return ans;  
}  
int main()  
{  
    int i,sum;  
    while(~scanf("%d",&n)&&n)  
    {  
        memset(vis,0,sizeof(vis));  
        memset(map,inf,sizeof(map));  
        int b,c,d,sta;  
        m=n*(n-1)/2;  
        for(i=0;i<m;i++)  
        {  
            scanf("%d%d%d%d",&b,&c,&d,&sta);  
            if(sta==0)  
                map[b][c]=map[c][b]=d;  
            else  
                map[b][c]=map[c][b]=0;  
        }  
        sum=prim();  
        printf("%d\n",sum);  
    }  
    return 0;  
}