Prim模板
#include<bits/stdc++.h> using namespace std; // T:确定属于最小生成树的节点集合 // S:剩余节点集合 int a[3010][3010]; int d[3010];// d[x], x表示x加入T时选出的最小边的权值 bool vis[3010];// vis[x]==true表示该节点已经加入了T集合 int n,m,ans=0; void prim() { memset(d,0x3f,sizeof(d)); memset(vis,0,sizeof(vis)); d[1]=0; for(int i=1; i<n; i++)//循环n-1次 每次找出一个点加入T { int x=0; for(int j=1; j<=n; j++)//遍历所有节点 { //当前d[]数组存储的是各个节点距离生成树的最小权值 if(!vis[j] && (x==0 || d[j]<d[x])) //!vis[j]表示该节点尚未加入T集合 x=j; //选出距离生成树最近的节点 } vis[x]=1;//将该节点加入到生成树中(T集合) for(int y=1; y<=n; y++) { if(!vis[y])//新加了一个点，就维护该点到还没有加入T集合的点的距离 d[y]=min(d[y],a[x][y]);//取距离生成树 } } } int main() { cin>>n>>m; //构建邻接矩阵 memset(a,0x3f,sizeof(a)); for(int i=1; i<=n; i++) a[i][i]=0; for(int i=1; i<=m; i++) { int x,y,z; cin>>x>>y>>z; a[x][y]=a[y][x]=min(a[x][y],z); } //求最小生成树 prim(); for(int i=2; i<=n; i++) ans+=d[i]; cout<<ans<<endl; return 0; } /* 输入 输出15 6 10 1 2 6 1 3 1 1 4 5 2 3 5 2 5 3 3 4 5 3 5 6 3 6 4 4 6 2 5 6 6 */