kuangbin专题六 HDU1233 还是畅通工程

题意:
中文题。
题解:
最小生成树板子题。

#include
#include
#include
using namespace std;
#define INF 0x3f3f3f3f
const int MAXN=105;
int map[MAXN][MAXN];
int dis[MAXN];
bool vis[MAXN];
int n,m;
void prim()
{
    int sum=0;
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++)
    dis[i]=map[1][i];
    for(int i=1;i<=n;i++)
    {
        int MIN=INF,k=0;
        for(int j=1;j<=n;j++)
        if(!vis[j]&&MIN>dis[j])
        MIN=dis[j],k=j;
        vis[k]=true;
        for(int j=1;j<=n;j++)
        if(!vis[j]&&map[k][j]map[k][j];
    }
    for(int i=1;i<=n;i++) sum+=dis[i];
    printf("%d\n",sum);
}
int main()
{
    while(~scanf("%d",&n),n)
    {
        memset(map,INF,sizeof(map));
        for(int i=1;i<=n;i++)
        map[i][i]=0;
        for(int i=1;i<=(n*(n-1))/2;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            if(w<map[u][v])
            map[u][v]=map[v][u]=w;
        }
        prim();
    }
} 

你可能感兴趣的:(最小生成树)