POJ 3723 Conscription

来源：互联网发布：录播软件编辑：程序博客网时间：2024/04/30 05:23

将问题做一下转换，如果两个人之间有关系就在两个人之间连一条边，边的权值是题目给的关系数值的相反数，假设没有关系的人之间的边的权值是0，那么问题就变成了在完全图里面找最小生成树，最小生成树的所有边的权值和加上（M+N）*10000就是最后的答案，当然没有必要构造出完全的最小生成树，如果包含关系的边用完了，就不用再用权值为0的边继续构造最小生成树了，因为这样对最小生成树的总权值和没有影响，用优先队列优化一下克鲁斯卡尔算法就OK了。

#include <stdio.h>#include <queue>using namespace std;             int merge_set[20005];int depth[20005];void merge(int root_a, int root_b){if(root_a < root_b)merge_set[root_a] = root_b;else{merge_set[root_b] = root_a;if(depth[root_a] == depth[root_b])depth[root_a]++;}}int get_root(int son){if(merge_set[son] == son)return son;return merge_set[son] = get_root(merge_set[son]);}typedef struct edge{int v1;int v2;int len;edge(int v1, int v2, int len){this->v1 = v1;this->v2 = v2;this->len = len;}}edge;priority_queue<edge> pq;bool operator < (const edge &a, const edge &b){return a.len > b.len;}void solve(int total_v){int total_cost, cnt;edge e(0,0,0);total_cost = 10000*total_v;cnt = 0;while(1){if(cnt == total_v-1)break;if(pq.empty())break;e = pq.top(); pq.pop();if(get_root(e.v1) == get_root(e.v2))continue;total_cost += e.len;cnt++;merge(get_root(e.v1), get_root(e.v2));}printf("%d\n", total_cost);}int main(void){int case_n, N, M, R, i;int male, fmale, value;//freopen("input.dat", "r", stdin);scanf("%d", &case_n);while(case_n--){scanf("%d %d %d", &N, &M, &R);for(i=0; i<M+N; i++){merge_set[i] = i;depth[i] = 1;}while(!pq.empty())pq.pop();while(R--){scanf("%d %d %d", &fmale, &male, &value);male += N;pq.push(edge(male, fmale, -1*value));}solve(M+N);}}

0 0