Kruskal算法
朴素版本:
1.用贪心的思想排序
2.用并查集判断边是否可选
3.用ans累加答案
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100010;
const int maxm=200010;
int n,m,f[maxn];//f是并查集
struct edge
{
int s,e,d;//起点 终点 长度
}ed[maxm];
bool cmp(edge a,edge b)
{
return a.d<b.d;
}
int get(int p)
{
if (f[p]==p) return p;
else return f[p]=get(f[p]);
}
int main()
{
scanf("%d%d",&n,&m);
for (int a=1;a<=m;a++)
scanf("%d%d%d",&ed[a].s,&ed[a].e,&ed[a].d);
sort(ed+1,ed+m+1,cmp);
for (int a=1;a<=n;a++)
f[a]=a;
int ans=0;
for (int a=1;a<=m;a++)
{
int s=ed[a].s,e=ed[a].e,d=ed[a].d;
int f1=get(s);
int f2=get(e);
if (f1!=f2)
{
f[f1]=f2;
ans+=d;
z[++k]=a;
}
}
printf("%d\n",ans);
return 0;
}
Prim算法
例题:
农民约翰被选为他们镇的镇长!他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。当然,他需要你的帮助。
约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。为了用最小的消费,他想铺设最短的光纤去连接所有的农场。
你将得到一份各农场之间连接费用的列表,你必须找出能连接所有农场并所用光纤最短的方案。
每两个农场间的距离不会超过100000
输入:
第一行: 农场的个数,N(3<=N<=100)。
第二行…结尾: 后来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个农场到它本身。
样例:
输入:
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
输出:
28
原理:
建立一个点集S,里面装着已经有边连接的点(初始只有节点1)
找出一条边,使它连接的端点恰好有一个点在S里且权值最小
答案ans+=这条边的权值
将另一个端点放进S里
计数器cnt++
当cnt==n(点数)-1时退出算法,输出ans
代码(原始版本,时间复杂度O(n^3)):
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
int a[110][110];
bool vis[110];
int b[110];
long long ans;
int o;
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++){
for (int j=1;j<=n;j++){
scanf("%d",&a[i][j]);
}
}
int s=1;
vis[s]=true;
b[++o]=s;
for (int i=1;i<n;i++){
int f1=0,f2=0,sum=114514;
for (int j=1;j<=o;j++){
for (int k=1;k<=n;k++){
if (b[j]!=k&&vis[k]==false&&sum>a[b[j]][k]){
sum=a[b[j]][k];
f1=b[j];
f2=k;
}
}
}
ans+=sum;
vis[f2]=true;
b[++o]=f2;
}
printf("%lld",ans);
return 0;
}
代码(正常版本,时间复杂度O(n^2+m)):
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n;
long long a[1100][1100],ans;
long long dis[1100];
bool vis[1100];
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++){
for (int j=1;j<=n;j++){
scanf("%lld",&a[i][j]);
if (i==j){
a[i][j]=1145141919810;
}
}
dis[i]=1145141919810;
}
vis[1]=true;
dis[1]=0;
for (int i=2;i<=n;i++){
dis[i]=min(dis[i],a[1][i]);
}
for (int i=1;i<n;i++){
long long sum=1145141919810;
int f=0;
for (int j=1;j<=n;j++){
if (vis[j]==false&&dis[j]<sum){
sum=dis[j];
f=j;
}
}
vis[f]=true;
ans+=sum;
for (int j=1;j<=n;j++){
if (j!=f)
dis[j]=min(dis[j],a[f][j]);
}
}
long long sum=0;
printf("%lld",ans);
return 0;
}