P1546 [USACO3.1]最短网络 Agri-Net(Prim)

P1546 [USACO3.1]最短网络 Agri-Nethttps://www.luogu.com.cn/problem/P1546 

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
 > que;
	que.push(P(0, 1));
	int nown = 0;
	while (!que.empty() && nown < n) {
		P t = que.top();
		que.pop();
		if (vis[t.second]) {
			continue;
		}
		vis[t.second] = true;
		ans += t.first;
		for (int i = 1; i <= n; i++) {
			if (i == t.second)
				continue;
			if (!vis[i] && edge[t.second][i] < d[i]) {
				d[i] = edge[t.second][i];
				que.push(P(d[i], i));
			}
		}
	}
}

int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			scanf("%d", &edge[i][j]);
		}
	}
	Prim();
	printf("%d", ans);
	return 0;
}