hdu1102 (prim模板题)

版权声明：本文为博主原创文章，转载请注明出处~~ https://blog.csdn.net/hxc2101/article/details/81937601

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1102

题意：有n个村庄需要修路使其连通，其中部分村落之间已经有路了，给出每两个村落之间修路的费用，求最小花费。

思路：给出的的数据就是临界矩阵的形式，考虑到了矩阵上无向图的"对称性"，直接输入到d[][]即可，对于已修路的村庄，缩成一点，遍历一下，是他们到其他村庄的代价花费取两者最小值即可。剩下的就是prim求最小生成树了。详见代码。

#include <bits/stdc++.h>
#pragma GCC optimize(3)
//#pragma GCC optimize("unroll-loops")
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC optimize("Ofast,no-stack-protector")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define pb push_back
#define mkp(a,b) make_pair(a,b)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define fi first
#define se second
#define lc (d<<1) //d*2
#define rc (d<<1|1) //d*2+1
#define eps 1e-9
#define dbg(x) cerr << #x << " = " << x << "\n";
#define mst(a,val) memset(a,val,sizeof(a))
#define stn(a) setprecision(a)//小数总有效位数
#define stfl setiosflags(ios::fixed)//点后位数:cout<<stfl<<stn(a);
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double PI=3.1415926535897932;
const int MAXN=1e5+10;
const ll mod=1e9+7;
ll inline mpow(ll a,ll b){ll ans=1;a%=mod;while(b){if(b&1)ans=(ans*a)%mod;a=(a*a)%mod,b>>=1;}return ans;}
int inline sgn(double x){return (x>-eps)-(x<eps);} //a<b:sgn(a-b)<0
priority_queue<int,vector<int>,greater<int> > qu; //up
priority_queue<int,vector<int>,less<int> > qd; //dn
const int inf = 0x3f3f3f3f; //9
const ll inff = 0x3f3f3f3f3f3f3f3f; //18

int n;
int d[110][110],vis[110],lw[110];
int q,a,b;
int ans=0;

int main()
{
    fio;
    while(cin>>n)
    {
        ans=0;
        mst(d,0x3f);mst(vis,0);mst(lw,0);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                cin>>d[i][j];
        cin>>q;
        for(int i=0;i<q;i++)
        {
            cin>>a>>b;
            for(int j=1;j<=n;j++)
                d[a][j]=d[b][j]=d[j][a]=d[j][b]=min(d[a][j],d[b][j]);
        }

        //prim
        for(int i=1;i<=n;i++)
            lw[i]=d[1][i];
        vis[1]=1;   //这句话一定要写！！！

        for(int i=1;i<=n-1;i++)
        {
            int mi=inf,p=-1;
            for(int j=1;j<=n;j++)
                if(!vis[j]&&mi>lw[j])
                {
                    p=j;
                    mi=lw[j];
                }
            vis[p]=1;
            ans+=mi;
            for(int j=1;j<=n;j++)
                if(!vis[j]&&lw[j]>d[p][j])
                    lw[j]=d[p][j];
        }
        cout<<ans<<endl;        
    }
}