PTA习题8.4 畅通工程之最低成本建设问题 (30分)

//c++,prim算法，无需并查集，把修路的成本看出边的权值就和最小生成树问题完全一样了

PTA习题8.4 畅通工程之最低成本建设问题 (30分)

某地区经过对城镇交通状况的调查，得到现有城镇间快速道路的统计数据，并提出“畅通工程”的目标：使整个地区任何两个城镇间都可以实现快速交通（但不一定有直接的快速道路相连，只要互相间接通过快速路可达即可）。现得到城镇道路统计表，表中列出了有可能建设成快速路的若干条道路的成本，求畅通工程需要的最低成本。

输入格式:
输入的第一行给出城镇数目N (1<N≤1000)和候选道路数目M≤3N；随后的M行，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号（从1编号到N）以及该道路改建的预算成本。

输出格式:
输出畅通工程需要的最低成本。如果输入数据不足以保证畅通，则输出“Impossible”。

#include <iostream>
#include<algorithm>
#include<vector>
#include<string>
#define MAXSIZE 1000000
using namespace std;
vector<int>visited;
vector<vector<int>>edges;
vector<int>d;//各顶点的当前最短
//prim 算法
int prim(int cur,int n)
{
	visited[cur] = 1;
	int sum = 0;
	for (int i = 0; i < n - 1; i++)
	{
		cur = -1;
		int min=MAXSIZE;
		for (int j = 0; j < n; j++)
		{
			if (visited[j] == 0 && d[j] < min)
			{
				cur = j;
				min = d[cur];
			}
		}
		if (cur == -1)return -1;
		visited[cur] = 1;
		sum += min;
		//更新d[]
		for (int v = 0; v < n; v++)
		{
			if (visited[v] == 0 && edges[cur][v] != 0 && edges[cur][v] < d[v]) {
				d[v] = edges[cur][v];
			}
		}
	}
	return sum;
	
}
int main() {
	//构图
	int n, m;//顶点数和边数
	cin >> n >> m;
	edges.resize(n);
	for (int i = 0; i < n; i++)
	{
		edges[i].resize(n);
	}
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			edges[i][j] = MAXSIZE;
			if (i == j)edges[i][j] = 0;
		}
	}
	for (int i = 0; i < m; i++)
	{
		int a, b,c;//输入位置和权数
		cin >> a >> b>>c;
		if (c < edges[a - 1][b - 1]) {
			edges[a - 1][b - 1] = c;
			edges[b - 1][a - 1] = c;
		}
	}
	//初始化辅助数组
	d.resize(n);
	for (int i = 0; i < n; i++)
	{
		d[i] = edges[0][i];
	}
	visited.resize(n);
	for (int i = 0; i < n; i++)
	{
		visited[i] = 0;
	}
	//调用函数
	int result = prim(0, n);
	if (result == -1)cout << "Impossible";
	else cout << result;
}