POJ 2349 （Prim算法模板）

1.模板代码

#include <iostream>
#include <cstring>
using namespace std;

const int SIZE=101;
const int INF=0x7fffffff;

int a[SIZE][SIZE];
int n, m;

void init()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            a[i][j]=INF;
}

void prim(int u0)
{
    int lowcost[SIZE], nearV[SIZE];
    int sum=0;

    for(int i=1;i<=n;i++) {lowcost[i]=a[u0][i]; nearV[i]=u0;}
    nearV[u0]=-1; lowcost[u0]=0;

    for(int i=1;i<n;i++)
    {
        int minV=INF, v=-1;
        for(int j=1;j<=n;j++)
        {
            if(nearV[j]!=-1 && lowcost[j]<minV)
            {
                minV=lowcost[j];
                v=j;
            }
        }
        cout<<nearV[v]<<" "<<v<<" "<<minV<<endl;
        nearV[v]=-1;
        sum+=minV;

        for(int j=1;j<=n;j++)
        {
            if(nearV[j]!=-1 && lowcost[j]>a[v][j])
            {
                lowcost[j]=a[v][j];
                nearV[j]=v;
            }
        }
    }

    cout<<"MST:"<<sum<<endl;
}

int  main()
{
    int u,v,w;

    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++){cin>>u>>v>>w; a[u][v]=a[v][u]=w;}
        cout<<"\n\n";
        prim(1);
    }

    return 0;
}

2.POJ 2349

题意:有P个点,用坐标给出,有两种联系方式:1每个点可以和距离在D以内的点相互联系,2有S个专门的卫星通道,两个点直接联系;
求D最小多少可以把这个图连起来
题解:首先不考虑S个卫星通道,先求最小生成树,用卫星通道把最小生成树中最大的S-1个边代替掉,然后剩下的最大的那条边的值就是能把整个图连起来的D的最小值.

#include <iostream>
#include <cmath>
#include <cstdio>
#include <algorithm>
using namespace std;

const int SIZE=505;

int S, P;
int x[SIZE], y[SIZE], cnt;
double w[SIZE][SIZE], pur[SIZE];

void init()
{
    for(int i=0;i<P;i++) w[i][i]=0x7fffffff;
    for(int i=0;i<P;i++)
        for(int j=i+1;j<P;j++)
        {
            double dd;
            dd=sqrt((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j]));
            w[i][j]=w[j][i]=dd;
        }
}

void prim(int u0)
{
    int nearV[SIZE];
    double lowcost[SIZE];

    cnt=0;
    for(int i=0;i<P;i++)
    {
        nearV[i]=u0;
        lowcost[i]=w[u0][i];
    }
    nearV[u0]=-1;

    for(int i=1;i<P;i++)
    {
        double minV=0x7fffffff;
        int u=-1;
        for(int j=0;j<P;j++)
        {
            if(nearV[j]!=-1 && minV>lowcost[j])
            {
                minV=lowcost[j];
                u=j;
            }
        }

        if(u==-1) {cout<<-1<<endl; return;}

        nearV[u]=-1;
        pur[cnt++]=minV;
        for(int i=0;i<P;i++)
        {
            if(nearV[i]!=-1 && lowcost[i]>w[u][i])
            {
                lowcost[i]=w[u][i];
                nearV[i]=u;
            }
        }
    }

    sort(pur,pur+cnt);
    printf("%.2f\n",pur[cnt-S]);
}

int main()
{
    int T;

    cin>>T;
    while(T--)
    {
        cin>>S>>P;
        for(int i=0;i<P;i++)
            cin>>x[i]>>y[i];
        init();
        prim(0);
    }

    return 0;
}