博客
关于我
并查集+线段树合并: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 DELETE 表别名问题
查看>>
MySQL Error Handling in Stored Procedures---转载
查看>>
MVC 区域功能
查看>>
MySQL FEDERATED 提示
查看>>
mysql generic安装_MySQL 5.6 Generic Binary安装与配置_MySQL
查看>>
Mysql group by
查看>>
MySQL I 有福啦,窗口函数大大提高了取数的效率!
查看>>
mysql id自动增长 初始值 Mysql重置auto_increment初始值
查看>>
MySQL in 太多过慢的 3 种解决方案
查看>>
MySQL InnoDB 三大文件日志,看完秒懂
查看>>
Mysql InnoDB 数据更新导致锁表
查看>>
Mysql Innodb 锁机制
查看>>
MySQL InnoDB中意向锁的作用及原理探
查看>>
MySQL InnoDB事务隔离级别与锁机制深入解析
查看>>
Mysql InnoDB存储引擎 —— 数据页
查看>>
Mysql InnoDB存储引擎中的checkpoint技术
查看>>
Mysql InnoDB存储引擎中缓冲池Buffer Pool、Redo Log、Bin Log、Undo Log、Channge Buffer
查看>>
MySQL InnoDB引擎的锁机制详解
查看>>
Mysql INNODB引擎行锁的3种算法 Record Lock Next-Key Lock Grap Lock
查看>>
mysql InnoDB数据存储引擎 的B+树索引原理
查看>>