SDU_week6_C - 掌握魔法の东东 I（prim最小生成树）

题目描述

东东在老家农村无聊，想种田。农田有 n 块，编号从 1~n。种田要灌氵
众所周知东东是一个魔法师，他可以消耗一定的 MP 在一块田上施展魔法，使得黄河之水天上来。他也可以消耗一定的 MP 在两块田的渠上建立传送门，使得这块田引用那块有水的田的水。 (1<=n<=3e2)
黄河之水天上来的消耗是 Wi，i 是农田编号 (1<=Wi<=1e5)
建立传送门的消耗是 Pij，i、j 是农田编号 (1<= Pij <=1e5, Pij = Pji, Pii =0)
东东为所有的田灌氵的最小消耗

Input
第1行：一个数n
第2行到第n+1行：数wi
第n+2行到第2n+1行：矩阵即pij矩阵

Output
东东最小消耗的MP值

Example
Input
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

Output
9

题目分析

本题要建立一棵最小生成树（或者多棵），但是考虑到以下两个问题：

1. 每棵生成树需要水源。
2. 每次要考虑使用水源的水还是建立田地之间连接。

我们把水源抽象成邻接矩阵的0号点，将Wi认为是pij[0][i]和pij[i][0]，从而得到一个（n+1）*（n+1）的矩阵。此矩阵完全避免了上述两个问题，直接对邻接矩阵进行prim求min_cost即可。

HITS

考虑到是稠密图，我们使用邻接矩阵+Prim算法；若稀疏图可以考虑结构体存边+Kruskal算法。

代码

#include <iostream>
#include<cstring>
using namespace std;

int pij[309][309];
int min_cost[309];//prim用于更新最小边集（已加入的点的最小边集）
bool vis[309];//prim中某点是否访问过

int main()
{
	memset(vis, 0, sizeof(vis));

    int n;//农田有 n 块;
    cin >> n;
	pij[0][0] = 0;//黄河之水天上来
	for (int i = 1; i <= n; i++)
	{//添加0号点
		cin >> pij[0][i];
		pij[i][0] = pij[0][i];
	}
	for (int i = 1; i <=n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			cin >> pij[i][j];
		}
	}

	//prim
	int start = 0;//种田开始的地方,水源
	vis[start] = true;
	int tot = 1;//已加入点集的点总数,总共n+1个，要加入n次
	int ans = 0;//cost之和
	for (int i = 0; i <= n; i++)min_cost[i] = pij[start][i];
	while (true)
	{
		int v = 0;
		int weight = 999999;
		for (int i = 0; i <=n; i++)
		{//选出一个没有经过且到点集距离最小的点
			if (!vis[i] && min_cost[i] < weight)
			{
				v = i;
				weight = min_cost[i];
			}
		}
		if (weight == 999999)break;//没有vis可用
		tot++;
		ans += min_cost[v];
		vis[v] = true;

		for (int i = 0; i <=n; i++)
		{//更新与v相连的边，如果更短的话
			if (!vis[i] && min_cost[i] > pij[v][i])
				min_cost[i] = pij[v][i];
		}
	}
	cout << ans;
	return 0;
}