题目链接
Prim算法:加点法。两个集合S和T,集合S是已经加入的集合,T是待加入的点的集合,每次在集合T中找距离集合S中的所有点距离最短(dis存储)的点,加入集合S,并用该点更新剩余点距离集合S点最近距离dis。
#include<iostream>
#include<algorithm>
#define MAXN 1010
#define inf 8989898
using namespace std;
int graph[MAXN][MAXN];
int dis[MAXN],vis[MAXN];
int main(){
int n,m;//1-n
cin>>n>>m;
//初始化
fill(graph[0],graph[0]+MAXN*MAXN,inf);
fill(dis,dis+MAXN,inf);
for(int i=0;i<m;i++){
int a,b,c;
cin>>a>>b>>c;
graph[a][b]=graph[b][a]=c;
}
int u=1;
dis[u]=0;
long long sum=0;
for(int i=1;i<=n;i++){
int minDis=inf,v=-1;
for(int j=1;j<=n;j++){
if(!vis[j]&&minDis>dis[j]){
v=j;
minDis=dis[j];
}
}
vis[v]=1;
sum+=dis[v];
for(int j=1;j<=n;j++){
if(!vis[j]&&dis[j]>graph[v][j]){
dis[j]=graph[v][j];
}
}
}
cout<<sum;
return 0;
}
Kruskal算法:加边法。利用并查集和排序。权重从小达到遍历,每次添加一条边(两端不连通),添加n-1次。
#include<iostream>
#include<algorithm>
#define MAXN 100010
using namespace std;
int p[MAXN];
void init(){
for(int i=0;i<MAXN;i++)p[i]=i;
}
int find(int a){
if(a!=p[a])p[a]=find(p[a]);
return p[a];
}
void merge(int a,int b){
int pa=find(a),pb=find(b);
if(pa!=pb)p[pb]=pa;
}
struct Edge{
int from,to,length;
};
struct Edge edge[MAXN];
bool cmp(Edge&a,Edge &b){
return a.length<b.length;
}
int main(){
init();
int n,m;//1-n
cin>>n>>m;
for(int i=0;i<m;i++){
int a,b,c;
cin>>edge[i].from>>edge[i].to>>edge[i].length;
}
sort(edge,edge+m,cmp);
long long sum=0;
int ct=0;
for(int i=0;i<m;i++){
if(find(edge[i].from)!=find(edge[i].to)){
merge(edge[i].from,edge[i].to);
sum+=edge[i].length;
ct++;
}
if(ct==n-1)break;
}
cout<<sum;
return 0;
}
意外收获:
以前没有注意fill和memset函数,这里总结一下。
fill(arr_start,arr_end,value)//数组中元素填充value,头文件是algorithm
memset(arr,value,sizeof(arr))//按照每个字节赋值arr,头文件是string.h
比较:由于memset按字节赋值的特性 memset事实上比 fill 快很多 ,节省了时间