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;}