最短路+最小生成树模板

最短路

Dijkstra算法

int a[105][105],d[105],n,m;
bool v[105];                   //d[j] -> 起点到j点的最短路
void dijkstra(){               //v[j] -> j点是否已经缩短过
	memset(v,0,sizeof(v));
	memset(d,0x3f,sizeof(d)); d[1]=0;
	for(int i=1;i<n;i++){
		int x=0;
		for(int j=1;j<=n;j++)
			if(!v[j]&&(x==0||d[j]<d[x]))x=j;
		v[x]=1; 		
		for(int y=1;y<=n;y++)
			d[y]=min(d[y],d[x]+a[x][y]);
	}
}

Floyd 算法:

int d[105][105];
int main()
{
	cin>>n>>m;
	memset(d,0x3f,sizeof(d));
	for(int i=1;i<=n;i++)d[i][i]=0;
	int x,y,z;
	for(int i=1;i<=m;i++){
		cin>>x>>y>>z;
		d[x][y]=min(d[x][y],z);
	}
	for(int k=1;k<=n;k++)
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				d[i][j]=min(d[i][j],d[i][k]+d[k]{j});
}

spfa算法（有负权值）

邻接表

struct edge{
	int v,w,next;
}edge[5010];
int head[5010],d[505],in[505];
bool vis[505];
int n,m,cnt; //cnt -> 边的编号
void add(int u,int v,int w)
{
      edge[++cnt].v=v; //edge[cnt].v -> cnt这条边的终点
      edge[cnt].w=w; //edge[cnt].w -> cnt这条边的权值
      edge[cnt].next=head[u]; 
                     //edge[cnt].next -> cnt这条边的上一条边的编号
      head[u]=cnt; //head[u] -> u作为起点的最后一条边的编号
}
bool spfa()
{
    memset(vis,0,sizeof(vis));
    memset(in,0,sizeof(in));
    memset(d,inf,sizeof(d));
    d[1]=0;vis[1]=1;in[1]++;
    queue<int>q;
    q.push(1);
    while(!q.empty()){
        int x=q.front();q.pop();
        vis[x]=0;
        for(int i=head[x];i!=-1;i=edge[i].next){
            int y=edge[i].v;
            if(d[y]>d[x]+edge[i].w){
                d[y]=d[x]+edge[i].w;
                if(!vis[y]){
                    q.push(y);vis[y]=1;in[y]++;
                    if(in[y]>=n)return false;//判断是否有负环
                }
            }
        }
    }
    return true;
}

 邻接矩阵

int a[310][310],d[310],n;
bool v[310];
void spfa(int m)
{
    queue<int>q;
    for(int i=1;i<=n;i++){
        if(i==m){
            d[i]=inf;
            v[i]=0;
        }
        else{
            d[i]=a[m][i];
            v[i]=1;
            q.push(i);
        }
    }
    while(!q.empty()){
        int x=q.front();
        q.pop();
        v[x]=0;
        for(int y=1;y<=n;y++){
            if(d[y]>d[x]+a[x][y]){
                d[y]=d[x]+a[x][y];
                if(!v[y]){
                    q.push(y);v[y]=1;
                }
            }
        }
    }
}

最小生成树

prim算法（邻接矩阵、稠密图）

int a[105][105],d[105],n,m,ans;
bool v[105];	         //v[i] -> i点是否在最小生成树中
void prim(){            //d[i] -> i点作为起点的边中的最小边
	memset(v,0,sizeof(v));
	memset(d,0x3f,sizeof(d));
	d[1]=0;
	for(int i=1;i<n;i++){
		int x=0;
		for(int j=1;j<=n;j++)
			if(!v[j]&&(x==0||d[j]<d[x]))
				x=j;
		v[x]=1;
		for(int y=1;y<=n;y++)
			if(!v[y])
				d[y]=min(d[y],a[x][y]);
	}
}
int main()
{
	cin>>n>>m;
	memset(a,inf,sizeof(a));
	for(int i=1;i<=n;i++)a[i][i]=0;
	int x,y,z;
	for(int i=1;i<=m;i++){
		cin>>x>>y>>z;
		a[x][y]=min(a[x][y],z);
	}
	prim();
	for(int i=2;i<=n;i++)ans+=d[i];
}

kruskal算法（无向图）

struct rec
{
    int x,y,z;
}edge[2000010];
int fa[2010],n,m,ans;
int ans;
bool operator<(rec a,rec b){
    return a.z<b.z;
}
int get(int x)
{
    if(x==fa[x])return x;
    return fa[x]=get(fa[x]);
}
int main()
{
    scanf("%d%d",&n,&m);
    int x[2005],y[2005];
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].z);
    sort(edge+1,edge+m+1);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=m;i++){
        int x=get(edge[i].x);
        int y=get(edge[i].y);
        if(x==y)continue;
        fa[x]=y;
        ans+=edge[i].z;
    }
    cout<<ans<<endl;
    return 0;
}