Uva1151——Buy or Build && POJ2784——Buy or Build

最小生成树的应用。

输入n个城市（1~n），和q个方案，接下来q行的第一个数为该方案中的城市数a，第二个数为用该方案需要的钱b，接下来a个数为城市的编号。接下来n行为第n个城市所在的坐标位置。

在这些城市之间建设道路，使得两两城市直接连通。

在两个城市之间建设道路，需要花费 两个城市之间的坐标的欧几里德距离，若买方案，则方案内的城市已经连通。

先进行一次求最小生成树，再枚举方案再求最小生成树，更新最小花费。

POJ上 的题目，只是输入方式不同。

还有一点最重要的是，用二进制枚举子集。

代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;

const int maxn = 1100;
int n, m, q, p[maxn];
vector<int> G[10];
int cost[10];

struct point
{
	int x, y;
}Point[maxn];
struct edge
{
	int start, end;
	int dist;
}Edge[maxn * maxn];


int get_dist(point a, point b)   //求两点之间的欧几里德距离
{
	return (a.x - b.x) * (a.x - b.x) + (a.y - b.y) * (a.y - b.y);
}
int cmp(edge a, edge b)     //按从小到大排序
{
	return a.dist < b.dist;
}

int finds(int x)   //并查集查找函数
{
	int r = x;
	while(p[r] != r)
		r = p[r];

	int i = x, j;
	while(i != r)
	{
		j = p[i];
		p[i] = r;
		i = j;
	}
	return r;
}

bool join(int x, int y)  //并查集合并函数
{
	int fx = finds(x);
	int fy = finds(y);
	if(fx != fy)
	{
		p[fx] = fy;
		return true;
	}
	return false;
}

int krus()   //求最小生成树
{
	int ans = 0;
	int num = 0;
	for(int i = 0; i < m && num < n - 1; i++)
	{
		if(join(Edge[i].start, Edge[i].end))
		{
			num++;
			ans += Edge[i].dist;
		}
	}
	return ans;
}

void solve()
{
	int i, j;
	for(i = 0; i <= n; i++)
		p[i] = i;
	int ans = krus();               //先求出不买方案，需要的花费。
	for(i = 0; i < (1 << q); i++)  //枚举子集
	{
		int co = 0;
		for(int kk = 0; kk <= n; kk++)
			p[kk] = kk;
		for(j = 0; j < q; j++)
		{	
			if(((i >> j) & 1))
				continue;
			co += cost[j];    
			for(int l = 1; l < G[j].size(); l++)   //合并方案内的城市为连通状态
				join(G[j][l], G[j][0]);
			
		}
		int an = krus();
		if(an + co < ans)
			ans = an + co;
	}
	cout << ans << endl;
}

int main()
{
//	freopen("1.txt", "r", stdin);
	int t, i, j, css = 1;
	cin >> t;
	while(t--)
	{
		if(css > 1)
			cout << endl;

		css++;
		for(i = 0; i < 10; i++)
			G[i].clear();

		scanf("%d%d", &n, &q);
		int k;
			
		for(i = 0; i < q; i++)
		{
			scanf("%d%d", &k, &cost[i]);
			int l;
			for(j = 0; j < k; j++)
			{
				scanf("%d", &l);
				G[i].push_back(l);
			}
		}
		for(i = 1; i <= n; i++)
			scanf("%d%d", &Point[i].x, &Point[i].y);
		m = 0;
		for(i = 1; i <= n; i++)   //两两城市直接建设道路的花费。
		{
			for(j = i + 1; j <= n; j++)
			{
				Edge[m].start = i;
				Edge[m].end = j;
				Edge[m++].dist = get_dist(Point[i], Point[j]);
			}
		}
		
		sort(Edge, Edge + m, cmp);
		solve();
	}
	return 0;
}