HDU 5919 Sequence II 主席树

题意:给定一个序列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 second
template<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("");
	}
	
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值