最小生成树问题

#include<bits/stdc++.h>
using namespace std;
#define il inline
/*
 Prim的思想是将任意节点作为根，再找出与之相邻的所有边（用一遍循环即可），
 * 再将新节点更新并以此节点作为根继续搜，
 * 维护一个数组：dis，作用为已用点到未用点的最短距离。

 * 证明：Prim算法之所以是正确的，
 * 主要基于一个判断：对于任意一个顶点v，
 * 连接到该顶点的所有边中的一条最短边(v, vj)必然属于最小生成树（即任意一个属于最小生成树的连通子图，
 * 从外部连接到该连通子图的所有边中的一条最短边必然属于最小生成树）
 */

/*
 * Kruskal
 * 先把边按照权值进行排序，用贪心的思想优先选取权值较小的边，并依次连接，
 * 若出现环则跳过此边（用并查集来判断是否存在环）继续搜，直到已经使用的边的数量比总点数少一即可。

证明：如果某个连通图属于最小生成树，
 那么所有从外部连接到该连通图的边中的一条最短的边必然属于最小生成树。
 当最小生成树被拆分成彼此独立的若干个连通分量的时候，所有能够连接任意两个连通分量的边中的一条最短边必然属于最小生成树
 */
il int read()
{
	 int x=0,f=1;char c=getchar();
	while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
	return x*f;
}//快读
#define inf 123456789
#define maxn 5005
#define maxm 200005
struct p
{
	int u,v,w;
}s[1010];
struct edge
{
	int v,w,next;
}e[maxm<<1];
//注意是无向图，开两倍数组
int f[1010];//并查集数组
typedef pair <int,int> pii;
priority_queue <pii,vector<pii>,greater<pii> > q;
int head[maxn];//head[i]表示以 i 为起点的最后一条边的编号
int dis[maxn];//已经加入最小生成树的的点到没有加入的点的最短距离，比如说1和2号节点已经加入了最小生成树，那么dis[3]就等于min(1->3,2->3)
int cnt,n,m,total,now=1,ans,cnt1;

bool vis[maxn];
//链式前向星加边
il void add(int u,int v,int w)
{
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}
bool com(p a,p b)
{
	return a.w<b.w;
}
int find(int x)//并查集
{
	return f[x]==x?x:f[x]=find(f[x]);
}
void krusal()
{
	cout<<"请输入城市的数量以及城市间能够建造通信线路的数量:"<<endl;
	n=read();
	m=read();
	cout<<"请输入能够建造通信线路的2个城市的序号以及它们的权值:"<<endl;
	for(int i=1;i<=n;i++)
		f[i]=i;
	for( int i=1;i<=m;++i)
	{
		s[i].u=read(),s[i].v=read(),s[i].w=read();

	}
	sort(s+1,s+1+m,com);
	for(int i=1;i<=m;i++)
	{
		int u=find(s[i].u);
		int v=find(s[i].v);
		if(u==v)
			continue;
		cout<<s[i].u<<" "<<s[i].v<<" "<<s[i].w<<endl;
		f[u]=v;
		ans+=s[i].w;
		total++;
		if(total==n-1)
			break;
	}

}
il int prim()
{
	cout<<"请输入城市的数量以及城市间能够建造通信线路的数量:"<<endl;
	n=read();
	m=read();
	cout<<"请输入能够建造通信线路的2个城市的序号以及它们的权值:"<<endl;

	for( int i=1,u,v,w;i<=m;++i)
	{
		u=read(),v=read(),w=read();
		add(u,v,w),add(v,u,w);
	}
	//先把dis数组附为极大值
	for( int i=2;i<=n;++i)
	{
		dis[i]=inf;
	}
	dis[1]=0;
	q.push(make_pair(0,1));
	while(!q.empty()&&cnt1<n)
	{
		int d=q.top().first,u=q.top().second;//从数组dis中选出离生成树最近的顶点 加入到生成树中（即在数组dis中找到最小值）
		q.pop();
		if(vis[u]) continue;
		cnt1++;
		ans+=d;
		vis[u]=1;
		for(int i=head[u];i;i=e[i].next)
			if(e[i].w<dis[e[i].v])
				dis[e[i].v]=e[i].w,q.push(make_pair(dis[e[i].v],e[i].v));//以顶点u为中间点,更新生成树到每一个非树顶点的距离
	}
	return ans;
}
int main()
{
	cout<<"1、克鲁斯卡尔求最小生成树"<<endl;
	cout<<"2、普利姆算法求最小生成树"<<endl;
	cout<<"3、退出"<<endl;
	while(true)
	{
		string a;
		cin>>a;
		if(a=="1")
		{
			ans=0;
			krusal();
			cout<<"构造此最小生成树的权值总和为"<<ans<<endl;
		} else if(a=="2"){
			ans=0;
			cout<<"构造此最小生成树的权值总和为"<<prim()<<endl;

		} else{
			break;
		}
	}


	return 0;
}
/*
 4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3
 */