最小生成树模板

 

普里姆算法

代码：

 

#include <iostream>
#include <iostream>
#include<algorithm>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<deque>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
int logo[1100],dis[1100],sum,map1[1100][1100];
int prim(int n)
{
    int i,j;
    for(i=0;i<=n;i++)
    {
        dis[i]=0x3f3f3f3f;
        logo[i]=0;
    }
    for(i=1;i<=n;i++)
    dis[i]=map1[1][i];
    logo[1]=1;
    dis[1]=0;
    for(i=1;i<n;i++)
    {
        int now=0x3f3f3f3f,min1=0x3f3f3f3f;
        for(j=1;j<=n;j++)
        {
            if(dis[j]<min1&&logo[j]==0)
            {
                min1=dis[j];
                now=j;
            }
        }
        if(now==0x3f3f3f3f)
        return 0;
        logo[now]=1;
        sum+=min1;
        for(j=1;j<=n;j++)
        {
            if(logo[j]==0&&dis[j]>map1[now][j])
            dis[j]=map1[now][j];
        }
    }
    return 1;
}
int main()
{
    int n,m,i,j;

    while(scanf("%d",&n)&&n)
    {

        memset(map1,0x3f3f3f3f,sizeof(map1));
         sum=0;
        for(i=0;i<n*(n-1)/2;i++)
        {
            int a,b,c;
            cin>>a>>b>>c;
            if(c<map1[a][b])
            map1[a][b]=map1[b][a]=c;
        }
        if(prim(n))
        cout<<sum<<endl;
        else
        cout<<"?"<<endl;

    }

} <iostream>
#include <iostream>
#include<algorithm>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<deque>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
int logo[1100],dis[1100],sum,map1[1100][1100];
int prim(int n)
{
    int i,j;
    for(i=0;i<=n;i++)
    {
        dis[i]=0x3f3f3f3f;
        logo[i]=0;
    }
    for(i=1;i<=n;i++)
    dis[i]=map1[1][i];
    logo[1]=1;
    dis[1]=0;
    for(i=1;i<n;i++)
    {
        int now=0x3f3f3f3f,min1=0x3f3f3f3f;
        for(j=1;j<=n;j++)
        {
            if(dis[j]<min1&&logo[j]==0)
            {
                min1=dis[j];
                now=j;
            }
        }
        if(now==0x3f3f3f3f)
        return 0;
        logo[now]=1;
        sum+=min1;
        for(j=1;j<=n;j++)
        {
            if(logo[j]==0&&dis[j]>map1[now][j])
            dis[j]=map1[now][j];
        }
    }
    return 1;
}
int main()
{
    int n,m,i,j;

    while(scanf("%d",&n)&&n)
    {

        memset(map1,0x3f3f3f3f,sizeof(map1));
         sum=0;
        for(i=0;i<n*(n-1)/2;i++)
        {
            int a,b,c;
            cin>>a>>b>>c;
            if(c<map1[a][b])
            map1[a][b]=map1[b][a]=c;
        }
        if(prim(n))
        cout<<sum<<endl;
        else
        cout<<"?"<<endl;

    }

}

 

 

 

 

 

克鲁斯卡尔算法:

代码：

 

#include <iostream>
#include <iostream>
#include<algorithm>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<deque>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
struct A{
    int a,b,c;
}a[1100];
int F[110000],r[110000],p[330000];
bool cmp(A a,A b)
{
    return a.c<b.c;
}
int find1(int n)
{
    int root=n;
    int temp=n;
    while(root!=F[root])
    root=F[root];
    while(temp!=root)
    {
        int t=F[temp];
        F[temp]=root;
        temp=t;
    }
    return root;
}
void union1(int x,int y)
{
    int root1=find1(x);
    int root2=find1(y);
    if(root1!=root2)
    {
        if(root1>root2)
        F[root1]=root2;
        else
        F[root2]=root1;
    }


}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)&&n)
    {
        for(int i=0;i<n;i++)
            scanf("%d%d%d",&a[i].a,&a[i].b,&a[i].c);
        sort(a,a+n,cmp);
        for(int i=0;i<=m;i++)
        F[i]=i;
        int k=0,x=0;
        for(int i=0;i<n;i++)
        {
            if(find1(a[i].a)!=find1(a[i].b))
            {
                x+=a[i].c;
                k++;
                union1(a[i].a,a[i].b);
            }

        }
        if(k==m-1)
        cout<<x<<endl;
        else
        cout<<"?"<<endl;
    }
} <iostream>
#include <iostream>
#include<algorithm>
#include<stdlib.h>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<deque>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
struct A{
    int a,b,c;
}a[1100];
int F[110000],r[110000],p[330000];
bool cmp(A a,A b)
{
    return a.c<b.c;
}
int find1(int n)
{
    int root=n;
    int temp=n;
    while(root!=F[root])
    root=F[root];
    while(temp!=root)
    {
        int t=F[temp];
        F[temp]=root;
        temp=t;
    }
    return root;
}
void union1(int x,int y)
{
    int root1=find1(x);
    int root2=find1(y);
    if(root1!=root2)
    {
        if(root1>root2)
        F[root1]=root2;
        else
        F[root2]=root1;
    }


}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)&&n)
    {
        for(int i=0;i<n;i++)
            scanf("%d%d%d",&a[i].a,&a[i].b,&a[i].c);
        sort(a,a+n,cmp);
        for(int i=0;i<=m;i++)
        F[i]=i;
        int k=0,x=0;
        for(int i=0;i<n;i++)
        {
            if(find1(a[i].a)!=find1(a[i].b))
            {
                x+=a[i].c;
                k++;
                union1(a[i].a,a[i].b);
            }

        }
        if(k==m-1)
        cout<<x<<endl;
        else
        cout<<"?"<<endl;
    }
}