最小生成树模板题POJ - 1287-prim+kruskal

POJ - 1287超级模板题

大概意思就是点的编号从1到N，会给你m条边，可能两个点之间有多条边这种情况，求最小生成树总长度?

这题就不解释了，总结就算，prim是类似dijkstra，从第一个点出发，每次走这个点没走过的最小边权值，这样不断找下去就可以找出，本质就是贪心算法

而kruskal是利用并查集，先按照边权值大小排序，然后从小的边开始往里面添加边，利用并查集判断是否在一个联通分量里面（就是是否相连）如果不相

连就建立边，从而建图，注意，节点编号如果是从1->n，那么相应初始化就应该从1->N；

prim版本：

 #include<iostream>
 #include<stdio.h>
 #include<string.h>
 using namespace std;
 const int INF = 0x3f3f3f3f;
 const int maxx = 1e2+;
 int cost[maxx][maxx];
 int mincost[maxx];
 bool used[maxx];
 int v;
 int m,n;
 int prim(){
   for (int i=;i<=n;i++){
     mincost[i]=INF;
     used[i]=false;
   }
  mincost[]=;//这里的数组的编号,代表我最开始建图的点
   int res=;
   while(true){
     int v = -;
     for (int u=;u<=n;u++){ //枚举边找到这个点的下一个边权值最小的点
         if (!used[u] && (v==- || mincost[u] < mincost[v])) v=u;
     }
     if (v == -)break;//没有的话就返回，无法建图
     used[v]=true;//顶点加入
     res+=mincost[v];
     for (int u=;u<=n;u++){
         mincost[u] = min(mincost[u],cost[v][u]);//？？？
     }
   }
   return res;
 }
 int main(){
   int tmp1,tmp2,tmp3;
    while(scanf("%d",&n) && n){
       scanf("%d",&m);
       if (m==){
         cout<<""<<endl;
         continue;
       }
      for (int i=;i<=n;i++){
         for(int j=;j<=n;j++){
             if (i==j)cost[i][j]=;
             else cost[i][j]=INF;
         }
      }
      for (int i=;i<=m;i++){
         scanf("%d%d%d",&tmp1,&tmp2,&tmp3);
         if (tmp3<cost[tmp1][tmp2])cost[tmp1][tmp2]=cost[tmp2][tmp1]=tmp3;//更新成最短边
      }
     cout<<prim()<<endl;
    }
   return ;
 }

Kruskal

#include<iostream>
#include<string.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node{
  int u,v,w;
}a[];
int p[];
int r[];
int n,m;
int find(int x){return p[x] == x ? x : p[x] = find(p[x]);}//找根节点
int cmp(node x,node y)
{
    return x.w<y.w;//按照边权值排序
}
int Kruskal(){
        int ans=;
        int num=;
        for (int i=; i<=n; i++)p[i]=i;
        sort(a+,a++m,cmp);
        for (int i=; i<=m; i++)
        {
            int x=find(a[i].u);//出发点的根节点
            int y=find(a[i].v);//到达点的根节点
            if (x!=y)//不是一个根节点
            {
                ans+=a[i].w;//连接
                p[x]=y;//y是x的父亲节点
                num++;//建立好了n-1条边
            }
            if (num==n-){
                break;
            }
        }
    return ans;
}
int main()
{
    int tmp1,tmp2,tmp3;
    while(~scanf("%d",&n))
    {
        if (n==)break;
        scanf("%d",&m);
        for (int i=; i<=m; i++)
        {
            scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
        }
        printf("%d\n",Kruskal());
    }
    return ;
}