B3603 [图论与代数结构 301] 最短树问题_1

题目描述
在离散数学课程的学习中，大家学习了最短树相关的知识，现在来检查一下大家的学习情况。

给定一张 nn 个点 mm 条边的带权连通无向图，点的编号从 11 到 nn，可能存在重边和自环。现在，请你求出这张图总边权和最小的支撑树（最短树）的边权之和。即，找到一颗支撑树，满足选用边的边权和最小，输出这个边权和。

输入格式
第一行两个正整数 n, mn,m ，分别表示给定无向图的点数和边数。

接下来 mm 行，每行三个非负整数 u, v, wu,v,w 表示一条连接 uu 和 vv 这两个点的无向边和这条边的边权 ww。

输出格式
输出仅一行一个非负整数，表示最短树的边权之和。

输入输出样例
输入 #1复制
4 5
1 3 1
1 2 2
2 3 0
3 4 3
1 4 8
输出 #1复制
4
说明/提示
【样例解释】

选择 (1,3)(1,3)、(2,3)(2,3) 和 (3,4)(3,4) 这三条边构成一颗支撑树。

【数据范围】

对于所有数据，满足 1\leq n \leq 20001≤n≤2000，1 \leq m \leq 30001≤m≤3000，边权均是 [0, {10}^9][0,10
9
] 中的整数，保证输入的图合法且满足题目中的限制条件。

请注意答案上界的大小，可能需要使用 C++ 中的 long long int 类型。

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=3010;
int g[N][N];
int dist[N];
bool st[N];
int n,m;

LL prim()
{
	memset(dist,0x3f,sizeof dist);
	
	LL 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;
		}
		
		if(i) res+=dist[t];
		
		st[t]=true;
		
		for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]);//更新点到集合树的距离
	}
	return res;
}

int main()
{
	cin>>n>>m;
	
	memset(g,0x3f,sizeof g);
	while(m--)
	{
		int a,b,c;
		cin>>a>>b>>c;
		g[a][b]=g[b][a]=min(g[a][b],c);
	}
	
	cout<<prim();
	
	return 0;
}