最小生成树邻接表模板

/*我的经验是prim加了堆优化之后比kruskal快一点点点点点点。所以99%的时间都是用的kruskal*/
/*prim算法（矩阵形式）：*/

#include<stdio.h>
#include<algorithm>
#define M 6000*3  //双向边 所以要乘 切记
#define inf 0x3f3f3f3f

int e;//边数  
//prim算法（边表形式）：

struct Edge//frm为起点,to为终点，w为边权，nxt指向下一个顶点
{
    int frm;
    int to,w,nxt;
}edge[M];

int vis[M],head[M],dis[M];
void addedge (int cu,int cv,int cw)//生成边的函数
{
    edge[e].frm = cu;
    edge[e].to = cv;
    edge[e].w = cw;
    edge[e].nxt = head[cu];//前一个.frm=cu的edge编号,如果没有 head[cu]=-1;
    head[cu] = e ++;//head[cu] 存最后一次存入的edge 编号
    
	edge[e].frm = cv;
    edge[e].to = cu;
    edge[e].w = cw;
    edge[e].nxt = head[cv];
    head[cv] = e ++;
}

int prim(int n,int sta) //n为顶点数量，sta为起点
{
    if(e/2<n-1)
		return -1; //边不够 返回-1 表示不完全图
    int sum = 0,i,flag;
    for(i=0;i<n;i++)
    {
	   dis[i]=inf;
    }
    memset(vis,0,sizeof(vis));
    for (i = head[sta];i != -1;i = edge[i].nxt)//遍历与sta点相连的所有顶点
    {
        int v = edge[i].to;
        dis[v] = edge[i].w;
    }
    vis[sta] = 1; //加入到最小生成树中
    int m = n - 1;  //只生成n-1条边，所以循环n-1次
	
    while (m --)
    {
	if(n==0)
		return-1
	if(e/2<n-1)
		return -1;
        int min = inf;
        for (i = 0;i < n;i ++)/出当前边权最小的边
		{
			if (!vis[i]&&dis[i] < min)
                flag = i,min = dis[i];
		}
		if(min==inf) //没找到可以用的边  代表图不完全 返回-1
			return -1;
        sum += dis[flag];
        vis[flag] = 1;//加入到最小生成树中
        for (i = head[flag];i != -1;i = edge[i].nxt)//更新与flag顶点相连的点的dis
        {                                           //
            int v = edge[i].to;
            if (edge[i].w < dis[v])
                dis[v] = edge[i].w;
        }
    }
    return sum; //返回边权总和
}

int main () 
{
	int a,b,w,n,i,j;
	while(scanf("%d",&n),n)
	{
		memset (head,-1,sizeof(head));
		e = 0;                  //记得初始化
		for(i=0;i<n*(n-1)/2;i++)
		{
				scanf ("%d%d%d",&a,&b,&w);
				a--,b--;
				addedge(a,b,w);
		}
		
		printf("%d\n",prim(n,1));
		//prim(int n,int sta) //n为顶点数量，sta为起点
	}
	return 0;
}



/*
邻接矩阵
int prim(int n,int sta)//n表示有n个顶点，sta表从sta这个顶点出发生成最小生成树
{
    int mark[M],dis[M],flag;
    int i,sum = 0,j;     //sum是总的最小生成树边权值
    for (i = 0;i < n;i ++) //初始化dis[i] 表从顶点sta到点i的权值
    {
        dis[i] = mat[sta][i];
        mark[i] = 0;
    }
    mark[sta] = 1;           //sta 这个顶点加入最小生成树中
    for (i = 1;i < n;i ++)   //循环n-1次，每次找出一条最小权值的边 n个点的图
    {                        //只有n-1条边
        int min = inf;       //inf 表无穷大
        for (j = 0;j < n;j ++)/出当前未在最小生成树中边权最小的顶点
            if (!mark[j] && dis[j] < min)
                min = dis[j],flag = j;
        mark[flag] = 1;         //把该顶点加入最小生成树中
        sum += dis[flag];       //sum加上其边权值
        for (j = 0;j < n;j ++)  //以falg为起点更新到各点是最小权值
            if (dis[j] > mat[flag][j])
                dis[j] = mat[flag][j];
    }
    return sum;       //返回边权总和
}
*/