最小生成树Prime算法

洛谷p1546链接
Prime算法的核心也是贪心，但是不同的就是，它是一直维护一颗树，
直到变成一颗最小生成树，

#include<bits/stdc++.h>
using namespace std;
const int maxn = 110;
const int inf = 0x3f3f3f3f;
int maze[maxn][maxn], dis[maxn], n;//maze数组存的是i ,j两条边的权重。
bool visit[maxn];//判断是否走过这个点//走过就标记为true,没走就是false
void Prime() {
	int ans = 0;//设置初始没有边为0，
	visit[1] = true;设置第一个点为走过，
	for(int i = 2; i <= n; i++)
		dis[i] = maze[1][i];//初始dis数组为后面n-1个点到1的最小值，
	for(int i = 1; i < n; i++) {
		int minn = inf ,u = 1;//设置当前最优值，
		for(int j = 1; j <= n; j++)//在mei'you
			if(!visit[j] && dis[j] < minn) {
				minn = dis[j];
				u = j;
			}
		ans += minn;//找到一条最优边，加上他的权重，
		visit[u] = true;
		for(int j = 1; j <= n; j++)//更新当前可以查询的最优边。
			if(!visit[j])
				dis[j] = min(dis[j],maze[u][j]);//比对这条边跟之前已经选过的点是否更好，更好就加入，不是就不加入，
	}
	cout << ans <<endl;
}
int main() {
	cin >> n;
	memset(visit,false,sizeof(visit));//初始化为false，都没有走过，
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= n; j++)
			cin >> maze[i][j];//输入每条边的权重，
	Prime();
	return 0;
}