ZOJ - 1203 Swordfish （最小生成树 Prim）

题目：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=203

题意：

n个点（xi，yi），将所有点相连的最小值

分析：

权值为距离的最小生成树，每两个都建立一条边，完全图的边数多，可以选择基于点数出发的Prim算法（当然，基于边数出发的Kruskal算法也可以）

代码：（Prim算法）

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <string>
#include <math.h>
#include <algorithm>
#include <queue>
#include <stack>
#include <vector>
#include <map>

using namespace std;

#define MAX 1000+10
#define MIN -1e+10
#define INF 0x7f7f7f7f

int t, n, m;

double Edge[MAX][MAX]; // 点间权值（距离）
double lowcost[MAX]; // （*） lowcost[i]：当前能够到达点i的最小权值（j到i最小，则lowcost[i] = j到i的距离）
double nearvex[MAX]; // （*） nearvex[i]：当前能够到达点i的最小值的起点坐标（j到i最小， 则nearvex[i] = j）

int Case;

void Prim()
{
	double ans = 0;
	int i, j;
	for(i = 1; i<=n; i++)
	{
		lowcost[i] = Edge[1][i]; // 默认起点为1，则到i点的最小距离，为1到i的距离
		nearvex[i] = 1;<span style="font-family: Arial, Helvetica, sans-serif;"> // 起点都为1，则距离i最近的点，都为1</span>
	}
	nearvex[1] = -1; // 把点1设为已达点，不再考虑
	for(i = 1; i<n; i++) // 找出n-1条边，连接n个点
	{
		int mint = INF, v = -1;
		for(j = 1; j<=n; j++) // 找出当前，由可达点出发的距离最小的点（如：第一次找出，由1出发到达其他未达点距离最小的点）
		{
			if(nearvex[j]!=-1 && lowcost[j] < mint) 
			{
				mint = lowcost[j];
				v = j;
			}
		}
		if(v == -1) break; // 存在回路
		ans += lowcost[v]; // 累加当前最优路径
		nearvex[v] = -1; // 把点v设为已达点，不在考虑
		for(j = 1; j<=n; j++)
		{
			if(nearvex[j]!=-1 && lowcost[j] > Edge[v][j]) // 维护由【可达点】到【未达点】的最小距离
			{
				lowcost[j] = Edge[v][j];
				nearvex[j] = v;
			}
		}
	}
	if(Case>1)printf("\n");
	printf("Case #%d:\n", Case++);
	printf("The minimal distance is: %.2lf\n", ans);
}

double x[MAX], y[MAX];

int main()
{
	int i, j;
	freopen("a.txt", "r", stdin);
	
	Case = 1;
	while(~scanf("%d", &n) && n)
	{
		memset(Edge, INF, sizeof(Edge));
		for(i = 1; i<=n; i++) scanf("%lf%lf", &x[i], &y[i]);
		for(i = 1; i<=n; i++)
		{
			for(j = i+1; j<=n; j++)
			{
				Edge[j][i] = Edge[i][j] = sqrt((x[i]-x[j])*(x[i]-x[j]) + (y[i]-y[j])*(y[i]-y[j]));
			}
		}
		Prim();
	}
	return 0;
}