ZOJ - 1586 QS Network （最小生成树 Kruskal）

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

题意：

T组数据，n个物体，n个物体的价格，矩阵为物体之间的距离，求所有物体相连之后的最小值（每个物体只能连一条线、如：再连1得重复购买该物体1）

分析：

最小生成树，权值不只是距离，得加上两端点物体的价格

代码：

#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 100000+10
#define MIN -1e+10
#define INF 0x7f7f7f7f

int t, n, m;

int parent[MAX*2+10];  // 并查集数组，判断连通性

struct edge
{
	int u, v, w;
	bool operator <(const edge& e)const
	{
		return w < e.w;
	}
}edges[MAX*2+10]; // 边权数组，权值排序 小--大

int  Find(int x) // 返回结点当前根结点 + 压缩路径
{
	int s;
	for(s = x; parent[s]>=0 ; s = parent[s]);
	while( s != x)
	{
		int temp = parent[x];
		parent[x] = s;
		x = temp;
	}
	return s;
}

void Union(int u, int v) // 合并两颗树 （根节点连和）
{
	int r1 = Find(u), r2 = Find(v); // 求出各自根节点
	int temp = parent[r1] + parent[r2]; // 根节点的双亲为负数，绝对值为该树的结点总数
	if(parent[r1] > parent[r2]) // 将结点数小的合并大结点数大的，（注意 值为负）
	{
		parent[r1] = r2;
		parent[r2] = temp;
	}
	else 
	{
		parent[r2] = r1;
		parent[r1] = temp;
	}
}

void Kruskal() // 格鲁斯卡算法实现
{
	int num = 0, ans = 0;
	memset(parent, -1, sizeof(parent));
	for(int i = 0; i<m; i++)
	{
		int u = edges[i].u;
		int v = edges[i].v;
		if(Find(u) != Find(v)) // 通过并查集，将非连通结点合并
		{
			ans += edges[i].w;
			num++;
			Union(u, v);
		}
		if(num>=n-1)break; // 全连通的边数为点数-1
	}
	printf("%d\n", ans);
}

int a[MAX*2+10];

int main()
{
	int i, j;
	//freopen("a.txt", "r", stdin);

	scanf("%d", &t);
	while(t--)
	{
		scanf("%d", &n);
		m = 0;
		int x, y;
		for(i = 1; i<=n; i++) scanf("%d", &a[i]);
		for(i = 1; i<=n; i++)
		{
			for(j = 1; j<=n; j++)
			{
				scanf("%d", &x);
				if(i>j) // 每条边只记录一次
				{
					edges[m].u = i;
					edges[m].v = j;
					edges[m].w = x+a[i]+a[j]; // 边权 = 距离 + 原件1 + 原件2
					m++;
				}
			}
		}
		sort(edges, edges+m);
		Kruskal();
	}
	return 0;
}