题目: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;
}