HDU 5919 Sequence II 主席树

来源:互联网 发布:在淘宝卖情趣用品 编辑:程序博客网 时间:2024/05/17 09:05

题意:给定一个序列nn,有mm次查询,每次查询一个区间[l,r][l,r],求区间中每一种数在区间中第一次出现的位置的中位数,强制在线。

如果考虑离线做法,从大到小枚举 rr,用线段树对每个 ll 维护[l,r][l,r]中数的种类数,查询时在线段树上二分,那么支持在线只需要简单可持久化一下,复杂度是 O((n+q) \log n)O((n+q)logn)


空间需要开到 2 * nlogn ,因为每次修改需要-1, 再 + 1,所以一共修改了两次,所以还有一个常数2。

代码: 

#include <cstdio>#include <cstring>#include <algorithm>#include <queue>#include <iostream>#include <string>#include <cmath>#include <vector>#include <set>#include <map>#include <bitset>#include <stack>using namespace std;#define REP(i,n) for ( int i=1; i<=int(n); i++ )  #define MP make_pair#define PB push_back#define SZ(x) (int((x).size()))#define ALL(x) (x).begin(), (x).end()#define X first#define Y secondtemplate<typename T> inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }template<typename T> inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }typedef long long LL;typedef long double LD;const int INF = 0x3f3f3f3f;template <class T>  inline bool RD(T &ret) {          char c; int sgn;          if (c = getchar(), c == EOF) return 0;          while (c != '-' && (c<'0' || c>'9')) c = getchar();          sgn = (c == '-') ? -1 : 1 , ret = (c == '-') ? 0 : (c - '0');          while (c = getchar(), c >= '0'&&c <= '9') ret = ret * 10 + (c - '0');          ret *= sgn;          return 1;  }  template <class T>  inline void PT(T x) {          if (x < 0) putchar('-') ,x = -x;          if (x > 9) PT(x / 10);          putchar(x % 10 + '0');}typedef pair<LL, LL> pii;const int N = 2e5 + 10;const int M = 30 * 2 * N;int ls[M], rs[M], root[N], tot, data[M];int new_node(int lst = 0) {ls[++ tot] = ls[lst];rs[tot] = rs[lst];data[tot] = data[lst];return tot;}void build(int l, int r, int &rt) {rt = new_node();if(l == r) return ;int m = (l + r) >> 1;build(l, m, ls[rt]);build(m + 1, r, rs[rt]);}void update(int pos, int val, int lst, int l, int r, int &rt) {rt = new_node(lst);data[rt] += val;if(l == r) return ;int m = (l + r) >> 1;if(pos <= m) update(pos, val, ls[lst], l, m, ls[rt]);else update(pos, val, rs[lst], m + 1, r, rs[rt]);}int query(int L, int R, int l, int r, int rt) {if(L <= l && r <= R) return data[rt];int m = (l + r) >> 1, ans = 0;if(L <= m) ans += query(L, R, l, m, ls[rt]);if(R > m) ans += query(L, R, m + 1, r, rs[rt]);return ans;}int ask_kth(int k, int l, int r, int rt) {if(l == r) return l;int m = (l + r) >> 1;return data[ls[rt]] >= k ? ask_kth(k, l, m, ls[rt]) : ask_kth(k - data[ls[rt]], m + 1, r, rs[rt]);}int a[N], pre[N];int main() {int T;cin >> T;for(int cas = 1; cas <= T; cas ++) {int n, m;scanf("%d %d", &n, &m);memset(pre, -1, sizeof(pre[0]) * (n + 5));for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);tot = 0;build(1, n, root[n + 1]);for(int i = n; i >= 1; i --) {if(~ pre[a[i]]) update(pre[a[i]], -1, root[i + 1], 1, n, root[i]);update(i, 1, root[~pre[a[i]] ? i : i + 1], 1, n, root[i]);pre[a[i]] = i;}printf("Case #%d:", cas);int ans = 0;while(m --) {int l, r;scanf("%d %d", &l, &r);l = (l + ans) % n + 1;r = (r + ans) % n + 1;if(l > r) swap(l, r);int cnt = query(l, r, 1, n, root[l]);ans = ask_kth((cnt + 1) >> 1, 1, n, root[l]);printf(" %d", ans);}puts("");}}


0 0
原创粉丝点击