Fork me on GitHub

P1546 最短网络(最小生成树)

题见洛谷

最小生成树

prim 并点

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string> 
#include<algorithm>
using namespace std;
int tu[101][101],dis[101],a[101];
int main()
{
    //freopen("a.in","r",stdin);              

    memset(tu,127,sizeof(tu));
    memset(dis,127,sizeof(dis));
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
     for(int j=1;j<=n;j++)
    {
        int c;
        scanf("%d",&c);
        if(c==0) c=0x7fffffff;
        tu[i][j]=c;
        tu[j][i]=c;

    }
    int tot=0;
    a[++tot]=1;
    for(int i=1;i<=n;i++) dis[i]=tu[1][i];
    dis[1]=0;

    int ans=0;
    while(tot<n)
    {
        int minl=0x7fffffff,mink;
        for(int i=1;i<=n;i++)
        {
            if(dis[i]<minl&&dis[i]) minl=dis[i],mink=i;
        }

        a[++tot]=mink;ans+=dis[mink];dis[mink]=0;

        for(int i=1;i<=n;i++)
         dis[i]=min(dis[i],tu[mink][i]);
    }
    //int ans=0;
    //for(int i=1;i<=n;i++)
    // ans+=dis[i];
    printf("%d",ans);
    return 0;
}

克鲁斯卡尔 并边

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string> 
#include<algorithm>
const int MAX=0x7fffffff;
using namespace std;
struct H{
    int x;
    int to;
    int v;
}a[101*101];


int v_comp(const H & a,const H & b)
{
    if(a.v<b.v) return 1;
    return 0;
}


int fa[200],b[200],n;
int findfather(int x)
{
    //if(fa[x]==x) return x;
    int k=x;
    while(fa[x]!=x)
     x=fa[x];
    return fa[k]=x;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) fa[i]=i;
    int cnt=0;
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++)
      {
          int k;
          scanf("%d",&k);
          if(k==0) k=MAX;
          a[++cnt].x=i;
          a[cnt].to=j;
          a[cnt].v=k;
      }

    sort(a+1,a+cnt+1,v_comp);

    int tot=0,ans=0;


    for(int i=1;i<=cnt;i++)
    {
        int f1=findfather(a[i].x),f2=findfather(a[i].to);
        if(f1!=f2) //一端在集合内,一端不在集合内 
        {
          //fa[a[i].x]=f1;fa[a[i].to]=f2;

          fa[f2]=f1;//路径压缩 
          ans+=a[i].v;
          tot++;
        }
        if(tot==n-1) break;
    }

    printf("%d",ans);
    return 0;
}
posted @ 2017-02-10 11:37  primes  阅读(160)  评论(0编辑  收藏  举报