博客
关于我
并查集+线段树合并:P3224-永无乡
阅读量:198 次
发布时间:2019-02-28

本文共 2094 字,大约阅读时间需要 6 分钟。

题目大意:

每个点有一个权值.并查集.在集合合并的过程中查询集合第k大.

题目思路:

并查集,每个点开个线段树.然后在并查集集合合并的过程中同时线段树合并即可.

AC代码:

#include
using namespace std;#define mid ((l + r) >> 1)const int maxn = 1e5 + 5;int sum[maxn << 5] , ls[maxn << 5] , rs[maxn << 5] , rt[maxn] , tot;void pushup (int t){ int tl = ls[t] , tr = rs[t]; sum[t] = sum[tl] + sum[tr];}int add (int l , int r , int t , int p , int c){ int now = ++tot; ls[now] = ls[t]; rs[now] = rs[t]; if (l == r) { sum[now] += c; return now; } if (p <= mid) ls[now] = add(l , mid , ls[now] , p , c); else rs[now] = add(mid + 1 , r , rs[now] , p , c); pushup(now); return now;}int ask (int l , int r , int t , int k){ if (l == r) return l; if (sum[ls[t]] >= k) return ask(l , mid , ls[t] , k); return ask(mid + 1 , r , rs[t] , k - sum[ls[t]]);}int mer (int a , int b , int l , int r){ if (!a) return b; if (!b) return a; if (l == r) { sum[a] += sum[b]; return a; } ls[a] = mer (ls[a] , ls[b] , l , mid); rs[a] = mer (rs[a] , rs[b] , mid + 1 , r); pushup(a); return a;}int a[maxn] , f[maxn] , n , m , id[maxn];int getf (int x){ return f[x]==x?x:f[x]=getf(f[x]);}int dsu_mer (int a , int b){ int fa = getf(a) , fb = getf(b); if (fa == fb) return false; f[fb] = fa; mer(rt[fa] , rt[fb] , 1 , n); return true;}int main(){ scanf("%d%d" , &n , &m); for (int i = 1; i <= n ; i++){ scanf("%d" , a + i); id[a[i]] = i; f[i] = i; rt[i] = add(1 , n , rt[0] , a[i] , 1); } for (int i = 1; i <= m ; i++){ int x , y; scanf("%d%d" , &x , &y); dsu_mer (x , y); } char op[5]; int x , y , q;scanf("%d" , &q); for (int i = 1; i <= q ; i++){ scanf("%s" , op); scanf("%d%d" , &x , &y); if (op[0] == 'Q'){ x = getf (x); if (sum[rt[x]] < y){ printf("-1\n"); }else { printf("%d\n" , id[ask(1 , n , rt[x] , y)]); } }else { dsu_mer (x , y); } } return 0;}

转载地址:http://gadi.baihongyu.com/

你可能感兴趣的文章
mysql中datetime与timestamp类型有什么区别
查看>>
MySQL中DQL语言的执行顺序
查看>>
mysql中floor函数的作用是什么?
查看>>
MySQL中group by 与 order by 一起使用排序问题
查看>>
mysql中having的用法
查看>>
MySQL中interactive_timeout和wait_timeout的区别
查看>>
mysql中int、bigint、smallint 和 tinyint的区别、char和varchar的区别详细介绍
查看>>
mysql中json_extract的使用方法
查看>>
mysql中json_extract的使用方法
查看>>
mysql中kill掉所有锁表的进程
查看>>
mysql中like % %模糊查询
查看>>
MySql中mvcc学习记录
查看>>
mysql中null和空字符串的区别与问题!
查看>>
MySQL中ON DUPLICATE KEY UPDATE的介绍与使用、批量更新、存在即更新不存在则插入
查看>>
MYSQL中TINYINT的取值范围
查看>>
MySQL中UPDATE语句的神奇技巧,让你操作数据库如虎添翼!
查看>>
Mysql中varchar类型数字排序不对踩坑记录
查看>>
MySQL中一条SQL语句到底是如何执行的呢?
查看>>
MySQL中你必须知道的10件事,1.5万字!
查看>>
MySQL中使用IN()查询到底走不走索引?
查看>>