http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=586
题目大意:
QS是一种生物,要完成通信,需要设备,每个QS需要的设备的价格不同,并且,这种设备只能在两个QS之间用一次,也就是说,如果一个QS需要和3个QS通信的话,它就必须得买3个设备,同时,对方三个也必须买对应的适合自己的设备。同时,每两个QS之间是有距离的,要完成通信还需要网线,给出每两个QS之间的网线的价值。求一棵生成树,使得所需要的费用最少。数据范围:所有数据都在1000以内。思路:
吃了好吃的继续~
好久没写prim了。来水一发。1A。
把邻接矩阵加上对应的接听器的值prim即可。
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int MAXN=1000+10; const int INF=9999999; int map[MAXN][MAXN],n,dis[MAXN],a[MAXN],vis[MAXN]; int prim() { for(int i=1;i<=n;i++) { dis[i]=INF; vis[i]=0; } int cur=1; vis[cur]=1; dis[cur]=0; for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) if(!vis[j] && map[cur][j] < dis[j]) dis[j]= map[cur][j] ; int mini=INF; for(int j=1;j<=n;j++) if(!vis[j] && mini > dis[j]) mini=dis[cur=j]; vis[cur]=true; } int sum=0; for(int i=1;i<=n;i++) sum+=dis[i]; return sum; } int main() { int T; scanf("%d",&T); while(T--) { scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { scanf("%d",&map[i][j]); map[i][j]+=a[i]+a[j]; } } int ans=prim(); printf("%d\n",ans); } return 0; }