博客
关于我
并查集+线段树合并: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 报错:Duplicate entry 'xxx' for key 'UNIQ_XXXX'
查看>>
Mysql 拼接多个字段作为查询条件查询方法
查看>>
mysql 排序id_mysql如何按特定id排序
查看>>
Mysql 提示:Communication link failure
查看>>
mysql 插入是否成功_PDO mysql:如何知道插入是否成功
查看>>
Mysql 数据库InnoDB存储引擎中主要组件的刷新清理条件:脏页、RedoLog重做日志、Insert Buffer或ChangeBuffer、Undo Log
查看>>
mysql 数据库中 count(*),count(1),count(列名)区别和效率问题
查看>>
mysql 数据库备份及ibdata1的瘦身
查看>>
MySQL 数据库备份种类以及常用备份工具汇总
查看>>
mysql 数据库存储引擎怎么选择?快来看看性能测试吧
查看>>
MySQL 数据库操作指南:学习如何使用 Python 进行增删改查操作
查看>>
MySQL 数据库的高可用性分析
查看>>
MySQL 数据库设计总结
查看>>
Mysql 数据库重置ID排序
查看>>
Mysql 数据类型一日期
查看>>
MySQL 数据类型和属性
查看>>
mysql 敲错命令 想取消怎么办?
查看>>
Mysql 整形列的字节与存储范围
查看>>
mysql 断电数据损坏,无法启动
查看>>
MySQL 日期时间类型的选择
查看>>