csp 201412-4 最优灌溉

裸的最小生成树

Prim：

//最小生成树
#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
#define maxn 1010

bool vis[maxn];
int cost[maxn][maxn];
int lowc[maxn];  //从已知点集到未知点的最小花费
int n,m;

int prim()
{
    int ans=0;
    memset(vis,false,sizeof vis);
    memset(lowc,inf,sizeof inf);
    vis[1]=true;
    for(int i=2;i<=n;i++)
        lowc[i]=cost[1][i];
    for(int i=1;i<n;i++)
    {
        int k=-1;
        int MIN=inf;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&lowc[j]<MIN)
            {
                MIN=lowc[j];
                k=j;
            }
        }
        if(k==-1) return -1;
        vis[k]=true;
        ans+=lowc[k];
        for(int j=1;j<=n;j++)
        if(!vis[j]&&lowc[j]>cost[k][j])
          lowc[j]=cost[k][j];
    }
    return ans;
}

int main()
{
    cin>>n>>m;
    memset(cost,inf,sizeof cost);
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        cost[u][v]=w;
        cost[v][u]=w;
    }
    int ans=prim();
    cout<<ans<<endl;
    return 0;
}

克鲁斯卡尔：

#include<bits/stdc++.h>
using namespace std;
#define maxn 1010
#define maxm 100010

struct Edge
{
    int u,v,w;
}edge[maxm];
int F[maxn];
int tot;

bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}

int Find(int x)
{
    if(F[x]==-1) return x;
    else return F[x]=Find(F[x]);
}

int kruskal(int n)
{
   memset(F,-1,sizeof F);
   sort(edge,edge+tot,cmp);
   int cnt=0;
   int ans=0;
   for(int i=0;i<tot;i++)
   {
       int u=edge[i].u;
       int v=edge[i].v;
       int w=edge[i].w;
       int t1=Find(u);
       int t2=Find(v);
       if(t1!=t2)
       {
           ans+=w;
           F[t1]=t2;
           cnt++;
       }
       if(cnt==n-1)  break;
   }
   if(cnt<n-1) return -1;
   else return ans;
}

void addedge(int u,int v,int w)
{
    edge[tot].u=u;
    edge[tot].v=v;
    edge[tot++].w=w;
}
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        addedge(u,v,w);
    }
    int ans=kruskal(n);
    cout<<ans<<endl;
    return 0;
}