2020长亭面试题D题 申请经费

Description

通过不懈努力，Alice 终于成为了 X 市网络安全办公室的负责人。作为一个务实的同学，Alice 立刻计划申请一批经费用于提高该市 N 家企业的网络安全水平，从而使每家企业符合新安全标准，新安全标准需要满足以下两个条件的任意一个:

1、有自己的安全团队
2、和任意一个满足标准的企业建立安全方面的合作关系
    假设最开始所有企业都没有自己的安全团队，相互之间也没有安全方面的合作关系。通过调研 Alice 知道了每家企业组建自己的安全团队需要补助的经费和一些企业之间合作需要补助的经费。现在需要你帮助 Alice 制定一个方案，使得所需要的补助的经费最少。

Input
第 1 行输入 1 个整数 TT (1≤T≤10)，表示一共有 T 组数据；

对于每一组数据:

  第 1 行输入 2 个整数 N M，( 1000001≤N≤1000,1≤M≤100000)，表示有 N 家企业，编号从 1 到 N，一共有 M 组合作关系；
  第 2 行输入 N 个整数 Wi(100001≤Wi≤10000)，第 i 个整数表示编号 i 的企业自己组建安全团队需要的补助；
  接下里 M 行，每行输入 3 个整数，a b C，表示编号 a 的企业和编号 b 的企业可以合作，合作需要的补助为 C (1≤C≤10000) (请注意，两个公司可能有多种合作方案)。

Output
    对于每组输入，输出一个整数，表示最少需要的补助。

#include 
#include 
#include 
using namespace std;
const int N = 1010, M = 1e6 + 10, INF = 0x3f3f3f3f;
int n, m;
int dist[N];
bool visit[N];
int map[N][N];
int prim()
{
	int res = 0;
	memset(dist, 0x3f, sizeof(dist));
	for (int i = 0; i <= n; i++)
	{
		int t = -1;
		for (int j = 0; j <= n; j++)
			if (!visit[j] && (t == -1 || dist[t] > dist[j]))
				t = j;
		if (i != 0 && dist[t] == INF) return INF;
		
		if (i != 0) res += dist[t];
		visit[t] = true;
		for (int j = 0; j <= n; j++) dist[j] = min(dist[j], map[t][j]);
	}
	return res;
}
int main()
{
	int T;
	cin >> T;
	while (T--)
	{
		cin >> n >> m;
		memset(map, 0x3f, sizeof(map));
		memset(visit, 0, sizeof(visit));
		for (int i = 1; i <= n; i++)
		{
			int x; cin >> x;
			map[0][i] = map[i][0] = min(map[0][i], x);
		}
		while (m--)
		{
			int a, b, w;
			cin >> a >> b >> w;
			map[a][b] = map[b][a] = min(map[a][b], w);
		}
		cout << prim() << endl;
	}
	return 0;
}