题目链接:点击打开链接
最小生成树裸题,就不讲述题意。主要是复习一下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;
}
比较简单,可以考虑做一些比较深入的最小生成树。