P3366 【模板】最小生成树

经过一番周折，终于差不多弄清了用堆优化的最小生成树的prim算法

这里先说几个原来未运用的知识点

pair

typedef pair<int,int> dui;

可以理解为一种结构体？？！！

有俩元素，可以为不同类型

 优先队列（堆）

priority_queue<int, vector<int> greater<int > > q; 这是最简单的一种形式

可以与pair 结合 

具体见代码

前向星式存图法，

题目：

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz

输入格式

第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）

接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式

输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

输入输出样例

输入 

4 5
1 2 2
1 3 2
1 4 3
2 3 4
3 4 3

输出 

7

说明/提示

时空限制：1000ms,128M

数据规模：

对于20%的数据：N<=5，M<=20

对于40%的数据：N<=50，M<=2500

对于70%的数据：N<=500，M<=10000

对于100%的数据：N<=5000，M<=200000

样例解释：

所以最小生成树的总边权为2+2+3=7

#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=5e3+10;
const int maxm=2e5+10;
const int inf=0x3f3f3f3f;
struct Edge
{
	int before;		//与这条边同一个起点但是再head中的前一条边的位置 
	int to; 		//终点是哪个点 
	int w;			//权重 
}edge[maxm<<1];		//无向图开两倍数组
int head[maxn],dis[maxm],vis[maxm],cnt,n,m,a,b,c,s,sum;
void add(int u,int v,int w)  //对边进行存储  前向星 
{
	edge[cnt].to=v;
	edge[cnt].w=w;
	edge[cnt].before=head[u];
	head[u]=cnt++; 
}
int main()
{
	cnt=0;
	memset(head,-1,sizeof head);
	memset(vis,0,sizeof vis);
	memset(dis,inf,sizeof dis);
	cin>>n>>m;			//	n个点  m条边 
	for(int i=0;i<m;i++)
	{
		cin>>a>>b>>c;
		add(a,b,c);
		add(b,a,c);	
	} 
	typedef pair<int,int> dui; //第一个位置防dis，爹第二个位置是点  
	priority_queue<dui,vector<dui>,greater<dui> > q; //优先队列只按第一个元素排序， 
	dis[1]=0;		//1号点到自己距离当然是零了 
	s=0;		//用来记录多少个点入了队列 
	sum=0;		//用来记录生成树共多长 
	q.push(make_pair(0,1));		//第一次先将0，1入队 
	while(!q.empty()&&s<n)		//队列不为空且入树顶点小于n时循环进行
	{
		int distance=q.top().first;
		int dot=q.top().second;
		q.pop();
		if(vis[dot])
		continue;
		s++;
		sum+=distance;
		vis[dot]=1;
		for(int i=head[dot];i!=-1;i=edge[i].before)   //注意 以前向星这种存储图的方式 只会遍历与当前顶点有关系的边 
		if(edge[i].w<dis[edge[i].to]) 
		{
			dis[edge[i].to]=edge[i].w; //这里就跟没有堆优化的一样，dis表示生成树到非数顶点的距离，如果比这个边的权值大，那么生 
			q.push(make_pair(dis[edge[i].to],edge[i].to));// 成数到这个点距离就是这个边的权值 
		}												//然后再将这个非树顶点到树的距离和这个点压入栈， 
	} 													不断地更新！ 
	if(s==n)
	cout<<sum<<endl;
	else
	cout<<"orz"<<endl;
//	int start;
//	cin>>start;
//	for(int i=head[start];i!=-1;i=edge[i].before)
//	cout<<start<<"->"<<edge[i].to<<" "<<edge[i].w<<endl;
	return 0;	
}