最小生成树模板prim和kruskal

题目链接

Prim算法：加点法。两个集合S和T，集合S是已经加入的集合，T是待加入的点的集合，每次在集合T中找距离集合S中的所有点距离最短(dis存储)的点，加入集合S，并用该点更新剩余点距离集合S点最近距离dis。

#include
#include
#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>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<

---

Kruskal算法：加边法。利用并查集和排序。权重从小达到遍历，每次添加一条边（两端不连通），添加n-1次。

#include
#include
#define MAXN 100010
using namespace std;
int p[MAXN];
void init(){
    for(int i=0;i>n>>m;
    for(int i=0;i>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

意外收获：

以前没有注意fill和memset函数，这里总结一下。

fill(arr_start,arr_end,value)//数组中元素填充value，头文件是algorithm

memset(arr,value,sizeof(arr))//按照每个字节赋值arr,头文件是string.h

比较：由于memset按字节赋值的特性  memset事实上比 fill 快很多 ，节省了时间