POJ1789Truck History最小生成树两种做法（Kruskal+Prim）模板题

就是题意不会太好理解和转换

最小生成树模板题

Kruskal：

//kruskal
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#include<cmath>
using namespace std;
#define inf 0x3f3f3f3f
#define inff 0x3f3f3f3f3f3f3f3f
const int N=2200;
#define mod 998244353
typedef long long ll;

int f[N];
char a[N][10];

struct node
{
    int l,r,d;
} e[N*N];

int getf(int x)
{
    if(f[x]==x)
        return x;
    return f[x]=getf(f[x]);
}

int merge(int x,int y)
{
    int t1=getf(x);
    int t2=getf(y);
    if(t1!=t2)
    {
        f[t2]=t1;
        return 1;
    }
    return 0;
}

bool cmp1(node x,node y)
{
    return x.d<y.d;
}
int main()
{
    ios::sync_with_stdio(false);
    int n;
    while(cin>>n)
    {
        if(n==0)
            break;
        memset(e,0,sizeof(e));
        /*for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(i==j)
                    e[i][j]=0;
                else
                    e[i][j]=inf;
            }
        }*/
        for(int i=1; i<=n; i++)
        {
            scanf("%s",a[i]);
            f[i]=i;
        }
        int p=0;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
                //for(int j=i+1; j<=n; j++)
            {
                int sum=0;
                for(int k=0; k<7; k++)
                {
                    if(a[i][k]!=a[j][k])
                        sum++;
                }
                //e[i][j]=e[j][i]=sum;
                e[p].l=i;
                e[p].r=j;
                e[p++].d=sum;
            }
        }
        int w,ans=0;
        sort(e,e+p,cmp1);
        for(int i=0; i<p; i++)
        {
            if(merge(e[i].l,e[i].r)==1)
            {
                w++;
                ans+=e[i].d;
            }
            if(w==n-1)
                break;
        }
        cout<<"The highest possible quality is 1/"<<ans<<"."<<endl;
    }
    return 0;
}

Prim：

//prim
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<map>
#include<cmath>
using namespace std;
#define inf 0x3f3f3f3f
#define inff 0x3f3f3f3f3f3f3f3f
const int N=2200;
#define mod 998244353
typedef long long ll;

int e[N][N],dist[N];
int n,ans;
bool book[N];
char a[N][10];

void prim()
{
    int countt=1;//countt代表点数，而不是边数
    ans=0;//记录路径长度
    for(int i=1;i<=n;i++)
    {
        dist[i]=e[1][i];
        book[i]=0;
    }
    book[1]=1;
    while(countt<n)
    {
        int minn=inf,u;
        for(int i=1;i<=n;i++)
        {
            if(!book[i]&&dist[i]<minn)
            {
                minn=dist[i];
                u=i;
            }
        }
        ans+=minn;
        countt++;
        book[u]=1;
        for(int i=1;i<=n;i++)
        {
            if(!book[i]&&dist[i]>e[u][i])
            {
                dist[i]=e[u][i];
            }
        }
    }

}
int main()
{
    ios::sync_with_stdio(false);
    while(cin>>n)
    {
        if(n==0)
            break;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(i==j)
                    e[i][j]=0;
                else
                    e[i][j]=inf;
            }
        }
        for(int i=1; i<=n; i++)
            scanf("%s",a[i]);
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
                //for(int j=i+1; j<=n; j++)
            {
                int sum=0;
                for(int k=0; k<7; k++)
                {
                    if(a[i][k]!=a[j][k])
                        sum++;
                }
                e[i][j]=e[j][i]=sum;
            }
        }
        prim();
        cout<<"The highest possible quality is 1/"<<ans<<"."<<endl;
    }
    return 0;
}