首页 > 代码库 > [2017年第0届浙江工业大学之江学院程序设计竞赛决赛 I] qwb VS 去污棒(并查集,按秩合并,最小生成树,LCA)
[2017年第0届浙江工业大学之江学院程序设计竞赛决赛 I] qwb VS 去污棒(并查集,按秩合并,最小生成树,LCA)
题目链接:http://115.231.222.240:8081/JudgeOnline/problem.php?cid=1005&pid=8
题意:中文题面。
手动画一下会发现所求边必然存在于最大生成树上,那么就可以首先构造一棵最大生成树。
问题转化成一棵树上求两个点之间的链上的最短边,用倍增lca就可以做了,但是我不会。
于是可以考虑建树时的操作,在求最大生成树的时候按秩合并,即集合大的根要做集合小的根的父亲,这样连一条有向边,保证路径上的所有边没有变化,并且能够维持整棵树高不会超过log(n)。
然后dfs预处理一下,查询的时候求lca,每次维护最短的边即可。
1 #include <bits/stdc++.h> 2 using namespace std; 3 4 typedef long long LL; 5 typedef struct Edge { 6 int u, v, w, next; 7 }Edge; 8 const int maxn = 50500; 9 const int maxm = 200200; 10 int h1[maxn], h2[maxn], pre[maxn], rk[maxn]; 11 int n, m, q, ecnt1, ecnt2; 12 int depth[maxn], dis[maxn]; 13 Edge e[maxm<<1], ee[maxn<<1]; 14 15 void init() { 16 for(int i = 1; i <= n; i++) pre[i] = i; 17 memset(h1, -1, sizeof(h1)); 18 memset(h2, -1, sizeof(h2)); 19 memset(rk, 0, sizeof(rk)); 20 memset(depth, 0, sizeof(depth)); 21 memset(dis, 0, sizeof(dis)); 22 ecnt1 = ecnt2 = 0; 23 } 24 25 void a1(int u, int v, int w) { 26 e[ecnt1].u = u, e[ecnt1].v = v, e[ecnt1].w = w; 27 e[ecnt1].next = h1[u], h1[u] = ecnt1++; 28 } 29 30 void a2(int u, int v, int w) { 31 ee[ecnt2].u = u, ee[ecnt2].v = v, ee[ecnt2].w = w; 32 ee[ecnt2].next = h2[u], h2[u] = ecnt2++; 33 } 34 35 bool cmp(Edge a, Edge b) { return a.w > b.w; } 36 int find(int x) { return x == pre[x] ? x : pre[x] = find(pre[x]); } 37 38 int unite(int x, int y, int w) { 39 x = find(x); y = find(y); 40 if(x != y) { 41 if(rk[x] >= rk[y]) { 42 if(rk[x] == rk[y]) rk[x]++; 43 pre[y] = x, a2(x, y, w); 44 } 45 else pre[x] = y, a2(y, x, w); 46 return 1; 47 } 48 return 0; 49 } 50 51 void dfs(int u, int p) { 52 for(int i = h2[u]; ~i; i=ee[i].next) { 53 int v = ee[i].v, w = ee[i].w; 54 if(p == v) continue; 55 depth[v] = depth[u] + 1; 56 dis[v] = w; pre[v] = u; 57 dfs(v, u); 58 } 59 } 60 61 int query(int u, int v) { 62 int ret = 0x7f7f7f7f; 63 if(depth[u] < depth[v]) swap(u, v); 64 while(depth[u] > depth[v]) { 65 ret = min(ret, dis[u]); 66 u = pre[u]; 67 } 68 while(u != v) { 69 ret = min(ret, min(dis[u], dis[v])); 70 u = pre[u]; v = pre[v]; 71 } 72 return ret; 73 } 74 75 int main() { 76 // freopen("in", "r", stdin); 77 int u, v, w; 78 while(~scanf("%d%d%d",&n,&m,&q)) { 79 init(); 80 for(int i = 0; i < m; i++) { 81 scanf("%d%d%d",&u,&v,&w); 82 a1(u, v, w); a1(v, u, w); 83 } 84 sort(e, e+ecnt1, cmp); 85 int mst = 0; 86 for(int i = 0; i < ecnt1; i++) { 87 u = e[i].u; v = e[i].v; w = e[i].w; 88 if(unite(u, v, w)) mst += w; 89 } 90 int rt = find(1); 91 memset(pre, -1, sizeof(pre)); 92 depth[rt] = 1; 93 dfs(rt, 0); 94 while(q--) { 95 scanf("%d%d",&u,&v); 96 printf("%d\n", query(u, v)); 97 } 98 } 99 return 0; 100 }
[2017年第0届浙江工业大学之江学院程序设计竞赛决赛 I] qwb VS 去污棒(并查集,按秩合并,最小生成树,LCA)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。