最小生成树模板

最小生成树的算法有两种，一种是prim，另一种是kruskal。

prim模板：

int prim()
{
    //low数组用来保存非生成树中各顶点与生成树中顶点最短边的权值；dis数组则是你输入的数组矩阵
    int ans = 0, i, j, flag, min;
    //ans用来存放最小生成树的总权值
    for(i = 1; i <= n; ++i)
    {
        low[i] = dis[1][i];//dis[1][i]，选取编号为1的顶点加入最小生成树中
        vis[i] = false;//数组vis，标记数组，用来表示顶点是否已加入最小生成树中。初始化为0，表示所有的点还未加入到最小生成树中
    }
    vis[1] = true;//编号为1的顶点加入最小生成树中
    for(i = 1; i <= n; ++i)
    {
        min = inf;//inf为无穷大，给min初始化
        flag = 1;
        for(j = 1; j <= n; ++j)
        {
            if(min > low[j] && !vis[j])//寻找满足一个顶点未加入生成树即未被访问过的点,另一个顶点已加入生成树的最小边
            {
                min = low[j];
                flag = j;
            }
        }
        if(min >= inf)
        {
            flag = -1;
            break;
        }
        ans += min;
        vis[flag] = true;//将顶点flag加入生成树
        for(j = 1; j <= n; ++j)
        {
            if(dis[flag][j] < low[j] && !vis[j])//更新由顶点flag到其他未加入生成树的顶点 边的权值
            {
                low[j] = dis[flag][j];
            }
        }
    }
    if(flag == -1)  return -1;
    return ans;
}

int prim()  
{  
    memset(vis , false , sizeof(vis));  
    int sum = 0;  
    for(int i = 0; i < n; i++) low[i] = map[0][i];  
    low[0] = 0;  
    vis[0] = true;  
    for(int i = 0; i < n-1; i++)  
    {  
        int x, m = oo;  
        for(int y = 0; y < n; y++) if(!vis[y] && low[y]<=m) m = low[x=y];  
        sum += m;  
        vis[x] = true;  
        for(int y = 0; y < n; y++) if(!vis[y] && low[y] > map[x][y]) low[y] = map[x][y];  
    }  
    return sum;  
}  

kruskal模板：

这个的话挺好理解，就是一个并查集

const int N = 1000;

struct EDG
{
    int v;
    int u;
    int w;
} edg[N];

int parent[N];

bool cmp(EDG a, EDG b)
{
    return a.w < b.w;
}

void set()
{
    for(int i = 0; i < N; i++)
        parent[i] = i;
}

int find_set(int n)
{
    int k, i, r = n;
    while(parent[r] != r)
    {
        r = parent[r];
    }
    k = n;
    while(k != r)
    {
        i = parent[k];
        parent[k] = r;
        k = i;
    }
    return r;
}

int kruskal(int n)
{
    int i, x, y, sum, cnt = 0;
    set();
    for(sum = 0, i = 0; i < n && cnt < n-1; i++)
    {
        x = find_set(edg[i].u);
        y = find_set(edg[i].v);
        if(x == y)    continue;
        parent[x] = parent[y];
        ++cnt;
        sum += edg[i].w;
    }
    return sum;
}