ZOJ 1586 QS Network（Prim 模板题）

/*
直接从第一个点开始放入 因为到最后这个点还是要被放入的
局部最优就能做到全局最优
从这个点开始向外延伸
想错了题意卡了好久。。。。
*/
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 1050;
const int INF = 0x3f3f3f3f;
int g[N][N];//储存点点之间的距离
bool st[N];
int dist[N];
int n;
int weight[N];//存储接口的价格

int prim()
{
	memset(dist, 0x3f, sizeof dist);//作用：后续找距离集合最近的点

	int res = 0;
	for (int i = 0; i < n; i++)//i仅仅表示迭代次数 n次 因为共有n个点(包括将第一个点放入集合)
	{
		int t = -1;
		for (int j = 1; j <= n; j++)//目的：从第一个点开始找起
		{
			if (!st[j] && (t == -1 || dist[t] > dist[j]))//由于t=-1的存在保证t的取值一定为正
				t = j;                                     //保证第一次迭代找到的第一个t一定为1且在st[j]下
		}
		if (i != 0 && dist[t] == INF) return INF;//if(i!=0) 即出现了不连通的情况 则没有最小生成树
		//因为t有可能是因为满足!st[j]和t==-1而产生的，他的dist可能等于INF
		if (i != 0) res += dist[t];//if(i!=0)
		st[t] = true;

		for (int j = 1; j <= n; j++) dist[j] = min(dist[j], g[t][j]);//一种取决与自身 另一种取决于此时的集合最近点
	}                          //因为是根据点到集合的定义：即为 点到集合内某个点的最短距离就是dist
							   //不断更新点到集合的距离 作用是：为了查询离集合最近的点
	return res;
}
int main()
{
	int t; cin >> t;
	while (t--)
	{
		ios::sync_with_stdio(false);
		cin.tie(0);
		cout.tie(0);
		memset(dist, 0x3f, sizeof dist);
		memset(g, 0x3f, sizeof g);
		memset(st, false, sizeof st);
		memset(weight, 0, sizeof weight);
		cin >> n;
		for (int i = 1; i <= n; i++)
			cin >> weight[i];

		//矩阵
		for (int i = 1; i <= n; i++)
		{
			for (int j = 1; j <= n; j++)
			{
				int c; cin >> c;
				if (i == j)g[i][j] = INF;//自身定义为无穷 其余点点距离取最短距离
				else
					g[i][j] = min(g[i][j], c + weight[i] + weight[j]);//继而简化为了普通的Prim
			}
		}
		cout << prim() << endl;
	}
}

小伙伴们 要期末考了呀 祝你们也能取得理想的成绩！