最小生成树题目

一、prim算法（稠密图）o(n^2)
思路：①找到集合外离集合最近的点，将它加到集合中；②更新其他点到集合的距离

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=505;
const int INF=0x3f3f3f3f;
int dist[N],g[N][N];
int n,m;
bool st[N];
int prim()
{
	memset(dist,INF,sizeof(dist));
	int res=0;
	for(int i=0;i<n;i++)
	{
		int t=-1;//t更新其他点到集合的距离
		for(int j=1;j<=n;j++)
		{
			if(!st[j]&&(t==-1||dist[t]>dist[j]))
				t=j;
		}
		if(i&&dist[t]==INF) return INF;
		if(i) res+=dist[t];//要在更新前加，因为存在自环的话，dist[t]会在更新时发生改变
		st[t]=true;
		for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);//更新集合外的点到集合的最短距离
	}
	return res;
}
int main()
{
	scanf("%d%d",&n,&m);
	memset(g,INF,sizeof(g));//先初始化为无穷大
	int u,v,w;
	for(int i=0;i<m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		g[u][v]=g[v][u]=min(g[u][v],w);//可能有重边的情况出现
	}
	int t=prim();
	if(t==INF)
			printf("impossible\n");
		else
			printf("%d\n",t);
	return 0;
}

二、Kruskal算法（稀疏图）O(mlogm)
思路：①将所有的边从小到大排序的；②枚举每条边顶点a,b,如果a,b不在同一集合里，将该边加入。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+10;
const int M=2e5+10,INF=0x3f3f3f3f;
int p[N];
int n,m;
struct s
{
	int u,v,w;
}edge[M];
bool cmp(s a,s b)
{
	return a.w<b.w;
}
int find(int x)
{
	if(x!=p[x])
		p[x]=find(p[x]);
	return p[x];
}
int kruskal()
{
	for(int i=1;i<=n;i++)//初始化并查集
		p[i]=i;
	sort(edge,edge+m,cmp);
	int res=0,cnt=0;
	for(int i=0;i<m;i++)
	{
		int a=find(edge[i].u),b=find(edge[i].v);
		if(a!=b)
		{
			p[a]=b;
			res+=edge[i].w;
			cnt++;//记录边的数量
		}
	}
	if(cnt!=n-1)
		return INF;
	else
		return res;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++)
		scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
	int t=kruskal();
	if(t==INF)//最后形成的树的边的数量不为n-1，那么就不是树了
		printf("impossible\n");
	else
		printf("%d\n",t);
	return 0;
}

1140. 最短网络
题意：把所有的村庄连通起来的最短路径。
思路：求最小生成树，它要求的除了是最短路径外，还要求所有的村庄要联通，所以是最小生成树。这里用prim算法比较方便，因为题目给出的是一个矩形，刚好是邻接矩阵，所以直接用prim算法来找点。然后这道题保证了一定是存在最小生成树的，所以可以不要判断不存在的情况，但要把dist[1]=0初始化好。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=110,INF=0x3f3f3f3f;
int dist[N],w[N][N];
int n;
bool st[N];
int prim()
{
	memset(dist,INF,sizeof(dist));
	dist[1]=0;//这里要先初始化好，第一个点是不计入路径的。
	int res=0;
	for(int i=0;i<n;i++)
	{
		int t=-1;
		for(int j=1;j<=n;j++)
		{
			if(!st[j]&&(t==-1||dist[t]>dist[j]))//找到集合距离最小的值
				t=j;
		}
		res+=dist[t];
		st[t]=true;
		for(int j=1;j<=n;j++)//更新所有点到集合的最小距离
			dist[j]=min(dist[j],w[t][j]);
	}
	return res;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			scanf("%d",&w[i][j]);
	int res=prim();
	printf("%d\n",res);
	return 0;
}

1141. 局域网
思路：用kruskal算法求出最小生成树的权值，用所有边的权值之和减去它既是答案。

注意，不能用if(x==y) res+=w;来计算被除去的边，因为当cnt==n-1时，有些边并未被计算过。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=110,M=210;
int p[N];
int n,k;
struct s
{
	int u,v,w;
}edge[M];
int cmp(s a,s b)
{
	return a.w<b.w;
}
int find(int x)
{
	if(x!=p[x])
		p[x]=find(p[x]);
	return p[x];
}
int kruskal()
{
	int res=0,cnt=0;
	for(int i=0;i<k;i++)
	{
		int x=find(edge[i].u),y=find(edge[i].v);
		if(x!=y)
		{
			p[x]=y;
			res+=edge[i].w;
			cnt++;
		}
		if(cnt==n-1)
			break;
	}
	return res;
}
int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)
		p[i]=i;
	long long sum=0;
	for(int i=0;i<k;i++)
	{
		cin>>edge[i].u>>edge[i].v>>edge[i].w;
		sum+=edge[i].w;
	}
	sort(edge,edge+k,cmp);
	int res=kruskal();
	cout<<sum-res<<endl;
	return 0;
}

1142. 繁忙的都市

#include<iostream>
#include<algorithm>
using namespace std;
const int N=310;
const int M=8010;
int p[N];
int n,m,s,maxn;
struct str
{
	int u,v,w;
}edge[M];
bool cmp(str a,str b)
{
	return a.w<b.w;
}
int find(int x)
{
	if(x!=p[x])
		p[x]=find(p[x]);
	return p[x];
}
int kruskal()
{
	for(int i=0;i<m;i++)
	{
		int u=find(edge[i].u),v=find(edge[i].v);
		if(u!=v)
		{
			p[u]=v;
			maxn=max(maxn,edge[i].w);
			s++;
		}
		if(s==n-1)
			break;
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		p[i]=i;
	for(int i=0;i<m;i++)
		scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
	sort(edge,edge+m,cmp);
	kruskal();
	printf("%d %d\n",s,maxn);
	return 0;
}

1143. 联络员

题意：有n个管理员通过直接或间接的方式要能够相互联系，其中有些通信通道是必须要选择的，求最小的通信费用。
思路：先将所有的必选通道合并为一个集合，同时累加所有的权值，再对可选通道进行选择，就是用最小生成树模板。

#include<iostream>
#include<algorithm>
using namespace std;
const int N=2010,M=10010;
int p[N];
int n,m,res;
struct str
{
	int u,v,w;
}edge[M];
int cmp(str a,str b)
{
	return a.w<b.w;
}
int find(int x)
{
	if(x!=p[x])
		p[x]=find(p[x]);
	return p[x];
}
int kruskal(int m)
{
	int cnt=0;
	for(int i=0;i<m;i++)
	{
		int u=find(edge[i].u),v=find(edge[i].v);
		if(u!=v)
		{
			p[u]=v;
			res+=edge[i].w;
		}
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		p[i]=i;
	int j=0;
	for(int i=0;i<m;i++)
	{
		int t,x,y,w;
		scanf("%d%d%d%d",&t,&x,&y,&w);
		if(t==1)
		{
			x=find(x),y=find(y);
			if(x!=y)
				p[x]=y;
			res+=w;
		}
		else
			edge[j].u=x,edge[j].v=y,edge[j++].w=w;
	}
	sort(edge,edge+j,cmp);
		kruskal(j);
		printf("%d\n",res);
	return 0;
}