学步园

题目链接：点击打开链接

最小生成树裸题，就不讲述题意。主要是复习一下MST。

Prim:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=103;
const int INF=0xffffff;
int map[N][N];
int n,Lowcost[N];
bool vis[N];

void Init(){
    for(int i=1;i<=n;i++)
    Lowcost[i]=INF;
    memset(vis,0,sizeof(vis));
}

int prim(int s){
    Init();vis[s]=true;
    int Sum=0;//初始化.
    
    for(int i=1;i<=n;i++)
    if(i!=s)//初始化Lowcost
    Lowcost[i]=map[s][i];

    for(int i=1;i<=n;i++){
       int PartLow=INF,p=s;
       //寻找到 目前的已合并集合 最短的节点
       for(int j=1;j<=n;j++)
       if(!vis[j]&&PartLow>Lowcost[j])
       {p=j;PartLow=Lowcost[j];}
       //判断
       if(p==s) return Sum;
       Sum+=PartLow;
       vis[p]=true;s=p;Lowcost[p]=INF;
       //更新Lowcost
       for(int j=1;j<=n;j++)
       if(!vis[j]&&map[s][j]<Lowcost[j])
       Lowcost[j]=map[s][j];
    }
}

int main(){
//    freopen("1.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++)
            scanf("%d",&map[i][j]);
        }
        printf("%d\n",prim(1));
    }
    return 0;
}

Kruscal:

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N=5005;
int father[N];
int n;
struct Rode{
    int x,y;
    int lenth;
}rode[N];

void Init(){
    for(int i=1;i<=n;i++)
    father[i]=i;
}

int find(int x){
    if(father[x]!=x)
    father[x]=find(father[x]);
    return father[x];
}

bool cmp(Rode x,Rode y){
    if(x.lenth<y.lenth) return true;
    else return false;
}

void Kruscal(){
    int Sum=0;
    Init();
    int rodenumber=(n*n-n)/2;
    sort(rode,rode+rodenumber,cmp);
    for(int i=0;i<rodenumber;i++){
        int a=find(rode[i].x);
        int b=find(rode[i].y);
        if(a!=b) {
            Sum+=rode[i].lenth;
            father[b]=a;
        }
    }
    printf("%d\n",Sum);
}

int main(){
    freopen("1.txt","r",stdin);
    while(scanf("%d",&n)!=EOF){
        int top=0,len;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                scanf("%d",&len);
                if(j>i){
                    rode[top].x=i;rode[top].y=j;
                    rode[top].lenth=len;
                    top++;
                }
            }
        }
        Kruscal();
    }
    return 0;
}

比较简单，可以考虑做一些比较深入的最小生成树。