Problm-A

概述：现在给出城镇数目，各个城镇之间的距离，以及已经修建道路的城镇。城镇之间的连通符合最小生成树。求最短需要修建多长的道路可以使城镇连通。

思路：用prim算法，取一个不再以连通的城镇，去权值，选择最小即可。最终输出为所求值。

感想：图都已经结课了。。上周因为一些选修课结课，需要写东西和复习，然后没刷题，这个周整整一周空闲，有机会了。然后这道题在提交时候前两次我选了图的基本算法.ppt提供的prim算法，那一段代码，在prim中定义的标记数组选了为1的时候，然后再本地编译器一直出错误，在oj上就一直ce，然后在网上查了一下，改成了数组为0，在判断时候为！，这样就过了？一样的意思，不知道为什么第一个不过。。。

#include<iostream>
#include<cstring>
#include<fstream>
#include<cstdio>

using namespace std;

const int INF = 1 << 20;
const int maxn = 105;
int N, Q, sum;
int dis[maxn];
int map[maxn][maxn];
bool sign[maxn];

void prim()
{
	sign[1] = 1;
	for (int i = 2;i <= N;i++)
		dis[i] = map[1][i];
	int now, min;
	for (int i = 1;i < N;i++)
	{
		min = INF;
		for (int j = 1;j <= N;j++)
			if (!sign[j] && dis[j] < min)
			{
				min = dis[j];
				now = j;
			}
		sum += min;
		sign[now] = 1;
		for (int j = 1;j <= N;j++)
			if (!sign[j] && map[now][j] < dis[j])
				dis[j] = map[now][j];
	}
}

int main()
{
	//ifstream cin("in.txt");
	while (cin >> N)
	{
		sum = 0;
		memset(sign, 0, sizeof(sign));
		for (int i = 1;i <= N;i++)
			for (int j = 1;j <= N;j++)
				cin >> map[i][j];
		cin >> Q;
		int a, b;
		for (int i = 1;i <= Q;i++)
		{
			cin >> a >> b;
			map[a][b] = map[b][a] = 0;
		}
		prim();
		cout << sum << endl;
	}
	return 0;
}

接下来是错误代码：

以为是头文件的原因，改了改头文件，但是显然不是。

#include<iostream>
#include<cstring>
#include<cmath>
#include<fstream>
#include<cstdio>
#include<algorithm>

using namespace std;

const int INF=1<<20;
const int maxn = 105;
int N, Q,sum;
int dis[maxn];
int map[maxn][maxn];
bool sign[maxn];

void prim()
{
	sign[1] = 0;
	for (int i = 2;i <= N;i++)
		dis[i] = map[1][i];
	int now, min;
	for (int i = 1;i < N;i++)
	{
		min = INF;
		for (int j = 1;j <= N;j++)
			if (sign[j] && dis[j] < min)
			{
				min = dis[j];
				now = j;
			}
		sum += min;
		sign[now] = 0;
		for (int j = 1;j <= N;j++)
			if (sign[j] && map[now][j] < dis[j])
				dis[j] = map[now][j];
	}
}

int main()
{
	ifstream cin("in.txt");
	while (cin >> N)
	{
		sum = 0;
		memset(sign, 0, sizeof(sign));
		for (int i = 1;i <= N;i++)
			for (int j = 1;j <= N;j++)
				cin >> map[i][j];
		cin >> Q;
		int a, b;
		for (int i = 1;i <= Q;i++)
		{
			cin >> a >> b;
			map[a][b] = map[b][a] = 0;
		}
		prim();
		cout << sum << endl;
	}
	return 0;
}