D - 继续畅通工程

3
1
0

分析：最短路径,可以用 prim算法解决，这个问题并不是很复杂，与之前我写的还是畅通工程差不多，只要你将已经修建好的路的成本不加进去就可以啦
AC代码：

#include <iostream>
#include<cstdio>
#include<cstring>
#define MAX 100001
using namespace std;
int g[101][101];
int lowcost[101];
int mincost;
int vis[101];//标记作用
void prim(int n)
{
    mincost=0;
    for(int i=1;i<=n;i++)
        lowcost[i]=g[1][i];
        vis[1]=1;
        int k;
        for(int i=1;i<n;i++)
        {
            int min=MAX;
            k=0;
            for(int j=1;j<=n;j++)
            {
                if(!vis[j]&&lowcost[j]<min)
                {
                    min=lowcost[j];
                    k=j;
                }
            }
            mincost+=min;
            vis[k]=1;
            for(int j=1;j<=n;j++)
                if(!vis[j]&&g[k][j]<lowcost[j])
                lowcost[j]=g[k][j];
        }
}


int main()
{
int n;
while(~scanf("%d",&n))
{
    if(n==0)  break;
    memset(vis,0,sizeof(vis));//清零
    for(int i=1;i<=n;i++)//邻接矩阵初始化
        for(int j=1;j<=n;j++)
    {
        if(i==j)
            g[i][i]=0;//自身到自身没有权值
        else g[i][j]=MAX;
    }
    int t;
    t=n*(n-1)/2;
    while(t--)
    {
        int a,b,c,d;
             scanf("%d %d %d %d",&a,&b,&c,&d);
             if(d==1)  c=0;
             g[a][b]=c;
             g[b][a]=c;
    }
    prim(n);
    printf("%d\n",mincost);
}
    return 0;
}