关于最小生成树（一）——prim算法

18448 最小生成树

时间限制:1000MS 代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题 语言: G++;GCC;VC
Description
给定结点数为n，边数为m的带权无向连通图G，所有结点编号为1,2,3…n。
求图G的最小生成树的边权和。

输入格式
第一行两个正整数n和m。n,m<=2000
之后的m行，每行三个正整数a,b,w，描述一条连接结点a和b,边权为w的边。1=<a,b<=n,w<=10^18。
注意可能存在重边和自环。

输出格式
一个整数表示图G的最小生成树的边权和（注意用长整型）。

输入样例
7 12
1 2 9
1 5 2
1 6 3
2 3 5
2 6 7
3 4 6
3 7 3
4 5 6
4 7 2
5 6 3
5 7 6
6 7 1

输出样例
16

#include<iostream>
#include<cstring>
#include<vector>
#include <limits.h>
using namespace std;
typedef long long ll;

ll n,m,i,j,k,x,y,z;

struct node
{
	ll adj,value;
}; 

vector<node> e[100000]; 

ll d[100000],v[100000]; 

ll getmin()
{
	ll minv=LLONG_MAX,mini,i;
	for(i=1;i<=n;i++)
	{
		if(v[i]==0&&d[i]<minv)
		{
			minv=d[i];
			mini=i;
		}
	}
	return mini;
}

void prim()
{
	memset(d,127,sizeof(d));
	d[1]=0;
	ll i,j,k,sum=0;
	for(i=1;i<=n;i++)
	{
		ll t=getmin();
		v[t]=1;
		sum+=d[t];
		for(j=0;j<e[t].size();j++)
		{
			ll adj=e[t][j].adj,value=e[t][j].value;
			if(d[adj]>value)
			{
				d[adj]=value;
			}
		}
	}
	cout<<sum;
}

int main(void)
{
	ios::sync_with_stdio(0),cin.tie(0);
	memset(v,0,sizeof(v));
	cin>>n>>m;
	for(i=1;i<=m;i++)
	{
		cin>>x>>y>>z;
		e[x].push_back({y,z}); 
		e[y].push_back({x,z}); 
	}
	prim();
	
	return 0;
}