prim模板 (最短路)

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

1.三步初始化！！！

2.注意 原先的图是否是连通的 若可能不连通需要判断 if(p==-1) 或 if(mi==inf) 的情况

/*
d[][]邻接矩阵 必须初始化为inf: memset(d,0x3f,sizeof(d))(inf每个字节都是0x3f)
lw[] 集合外的点i 到集合内的点的最小代价
vis[] 记录该点是否已经放入集合
*/
int n;
int d[30][30];
int vis[30];
int lw[30];
int ans=0;
 
int main()
{
    fio;
    while(cin>>n&&n)
    {
        ans=0;memset(vis,0,sizeof(vis));memset(d,0x3f,sizeof(d));   //初始化 这三步必须有
        
        //...此处省略输入邻接矩阵的过程  注意因为无向图 d[a][b]=val;d[b][a]=val; 要写两遍
        
        for(int i=1;i<n;i++)    //因为所有点都会在树中,所以这里任意选一个点即可 (先往树中放入点0)
            lw[i]=d[i][0];
        vis[0]=1;      //这句话一定要写！！！！！
 
        //prim
        for(int i=1;i<n;i++)  //这个for表示加入n-1条边
        {
            int mi=inf,p=-1;
            for(int j=0;j<n;j++)
                if(!vis[j] && mi>lw[j])
                {
                    mi=lw[j];
                    p=j;
                }
            if(p==-1) break;  //(因为本题不需要) 如果p==-1或mi==inf 表示原图不通
            vis[p]=1;
            ans+=mi;
            for(int j=0;j<n;j++)
                if(!vis[j] && lw[j]>d[p][j]) //update
                    lw[j]=d[p][j];
        }
        cout<<ans<<endl;
    }
    return 0;
}

 

题目代码：

#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

/*
d[][]邻接矩阵 必须初始化为inf: memset(d,0x3f,sizeof(d))(inf每个字节都是0x3f)
lw[] 集合外的点i 到集合内的点的最小代价
vis[] 记录该点是否已经放入集合
*/


int n;
int d[30][30];
int vis[30];
int lw[30];
int ans=0;

int main()
{
    fio;
    while(cin>>n&&n)
    {
        ans=0;mst(vis,0);mst(d,0x3f);
        for(int i=0;i<n-1;i++)
        {
            char ch;
            int m;
            cin>>ch>>m;
            while(m--)
            {
                int p;
                cin>>ch>>p;
                d[i][ch-'A']=p;
                d[ch-'A'][i]=p;
            }
        }

        for(int i=1;i<n;i++)
            lw[i]=d[i][0];
        vis[0]=1;

        //prim
        for(int i=1;i<n;i++)
        {
            int mi=inf,p=-1;
            for(int j=0;j<n;j++)
                if(!vis[j] && mi>lw[j])
                {
                    mi=lw[j];
                    p=j;
                }               //(因为本题不需要) 如果p==-1或mi==inf 表示原图不通
            vis[p]=1;
            ans+=mi;
            for(int j=0;j<n;j++)
                if(!vis[j] && lw[j]>d[p][j]) //update
                    lw[j]=d[p][j];
        }
        cout<<ans<<endl;
    }
    return 0;
}