hdu 1879——继续畅通工程 最小生成树

hdu 1879 点击打开链接

若状态为1 则将两点归并

克鲁斯卡尔 算法：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
using namespace std;

int father[110],map[110][110];
int ans,n;
struct node
{
    int x,y;
    int c;
} s[5100];

bool cmp(node a,node b)
{
    return a.c<b.c;
}
int find(int i)
{
    if(father[i]!=i)
        father[i]=find(father[i]);
    return father[i];
}
void merge(int a,int b)
{
    a=find(a);b=find(b);
    if(a>b)
      father[a]=b;
    else
        father[b]=a;
}
void kru()
{
    for(int i=0; i<n*(n-1)/2; i++)
    {
        int a=find(s[i].x);
        int b=find(s[i].y);
        a=father[a];//使得father节点都为一个节点。
        b=father[b];
        if(a!=b)
        {
            ans+=s[i].c;
            merge(a,b);
        }
    }
}
int main()
{
    int i,z,tem;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
            break;
        for(i=0; i<=n; i++)
            father[i]=i;

        for(i=0; i<n*(n-1)/2; i++)
        {
            scanf("%d%d%d%d",&s[i].x,&s[i].y,&s[i].c,&z);
            if(z==1)//在输入过程中即更新father数组
            {
                int xx=find(s[i].x);
                int yy=find(s[i].y);
                if(xx!=yy)
                    merge(xx,yy);
            }
        }
        sort(s,s+n*(n-1)/2,cmp);
        ans=0;
        kru();
            printf("%d\n",ans);
    }
    return 0;
}

prim算法：

#include<stdio.h>
#include<string.h>
int map[110][110],visit[11000],low[110];
int n,ans;
void prim()
{
    int pos,i,j,min=999999;
    visit[1]=1;pos=1;
    for(i=1;i<=n;i++)
    {
        if(i!=pos)
            low[i]=map[pos][i];
    }
    for(i=1;i<n;i++)
    {
        min=99999999;
        for(j=1;j<=n;j++)
        {
            if(visit[j]!=1&&min>low[j])
            {
                min=low[j];
                pos=j;
            }
        }
        visit[pos]=1;
        ans+=low[pos];
        for(j=1;j<=n;j++)
        {
            if(visit[j]!=1&&map[pos][j]<low[j])
                low[j]=map[pos][j];
        }
    }
}

int main()
{
    int x,y,c,i,z;
    while(scanf("%d",&n)!=EOF)
    {
        if(n==0)
            break;
        memset(map,0,sizeof(map));
        memset(visit,0,sizeof(visit));
        for(i=0;i<n*(n-1)/2;i++)
        {
            scanf("%d%d%d%d",&x,&y,&c,&z);
            map[x][y]=map[y][x]=c;
            if(z==1)
                map[x][y]=map[y][x]=0;//key：：不能用visit标记 否则在对low数组更新是会出问题！！！
        }
        ans=0;
        prim();
        printf("%d\n",ans);
    }
    return 0;
}