prim 算法加模板

最小生成树的prim算法：

1，在顶点中任取一点，加入集合U中；

2，在集合外的点中找到离集合中的点中最近的点并构成路径，找到的点加入集合中；

3，重复2过程到树构建完毕；

举个栗子：

从1开始，1加入集合U中，2,3,4,在集合外；

选择离集合中点最近的点4,4加入集合中，2,3在集合外；

选择离集合中点最近的点2,2加入集合中，3在集合外；

选择离集合中点最近的点3,3加入集合中，完毕；

测试：

代码：

#include<iostream>#include<cstring>#include<climits>using namespace std;const int M=500;int cost[M][M];int Creat(){int n,m;cin>>n>>m;for(int q=1;q<=n;q++){for(int w=1;w<=n;w++)cost[q][w]=INT_MAX;}int a,b,c;for(int q=1;q<=m;q++){cin>>a>>b>>c;cost[a][b]=c;cost[b][a]=c;}return n;}void prim(int n){int close[M],lowcost[M];for(int q=1;q<=n;q++){close[q]=1;lowcost[q]=cost[1][q];}close[1]=-1;for(int q=1;q<n;q++){int Min=INT_MAX,u=q;for(int w=1;w<=n;w++)if(close[w]!=-1 && lowcost[w]<Min){Min=lowcost[w];u=w;}cout<<"("<<close[u]<<","<<u<<")"<<" "<<lowcost[u]<<endl;for(int w=1;w<=n;w++)if(close[w]!=-1 && lowcost[w]>cost[u][w]){lowcost[w]=cost[u][w];close[w]=u;}}return;}int main() {int t;cin>>t;while(t--){int n=Creat();prim(n);}return 0;}