每日一题 No.55 Conscripttion

本题要求：

需要招募女兵N人，男兵M人。每征募一个人需要花费10000美元。但是如果已经招募的人中有一些关系亲密的人，那么可以少花一些钱。给出若干的男女之间的1~9999之间的亲密度关系，征募某个人的费用是10000-（已经征募的人中和自己的亲密度的最大值）。要求通过适当的征募顺序使得征募所有人所需费用最小。

输入格式：

第一行输入 N M R（关系数）
接下来R行
每行输入f t c 代表第f号男与第t号女的亲密度为c

输出格式：

输出最小费用

输入样例：

5 5 8
4 3 6831
1 3 4583
0 0 6592
0 1 3063
3 3 4975
1 3 2049
4 2 2104
2 2 781

输出样例：

71071

解题思路 :

使用kruskal算法算法即可

代码 :

#include <iostream>#include <vector>#include <algorithm>using namespace std;  class Edge {    public:        int u;        int v;        int cost;        Edge(int u, int v, int cost) {            this->u = u;            this->v = v;            this->cost = cost;        }        Edge() {        }};using namespace std;  int par[1001];int rank[1001];void init(int n) {    for (int i = 0; i < n; i++) {        par[i] = i;        rank[i] = 0;    }}int find(int x) {    if (par[x] == x) {        return x;    } else {        return par[x] = find(par[x]);    }}void unite(int x, int y) {    x = find(x);    y = find(y);    if (x == y) {        return;    } else if (rank[x] < rank[y]){        par[x] = y;    } else {        par[y] = x;        if (rank[x] == rank[y]) {            rank[x]++;        }    }}bool same(int x, int y) {    return find(x) == find(y); } bool comp(const Edge& e1, const Edge& e2) {    return e1.cost < e2.cost;}vector<Edge> es;int V, E;int kruskal() {    sort(es.begin(), es.end(), comp);    init(V);    int res = 0;    for (int i = 0; i < E; i++) {        Edge e = es[i];        if (!same(e.u, e.v)) {            unite(e.u, e.v);            res += e.cost;        }    }    return res;}int main() {    int N, M, R;    cin >> N >> M >> R;    V = N + M;    E = R;    for (int i = 0; i < R; i++) {        int f, t, c;        cin >> f >> t >> c;        es.push_back(Edge(f, t + N, -c));    }    cout << 10000 * V + kruskal() << endl;    return 0; }