「luogu2617」Dynamic Rankings

「luogu2617」Dynamic Rankings

传送门
树套树直接上树状数组套主席树,常数很大就是了。
树套树参考代码:

/*--------------------------------
  Code name: DynamicRanking.cpp
  Author: The Ace Bee
  This code is made by The Ace Bee
--------------------------------*/
#include <cstdio>
#include <cstring>
#include <algorithm>
#define rg register
using namespace std;
const int MAXN = 500010;
inline int lowbit(int x) { return x & -x; }
inline int read() {
	int s = 0; bool f = false; char c = getchar();
	while (c < '0' || c > '9') f |= (c == '-'), c = getchar();
	while (c >= '0' && c <= '9') s = (s << 3) + (s << 1) + (c ^ 48), c = getchar();
	return f ? -s : s;
}
int n, m, a[MAXN], len, b[MAXN], rt[MAXN];
int tot, num[MAXN << 5], lc[MAXN << 5], rc[MAXN << 5];
int cnt[2], tmp[2][20];
struct node{ bool f; int l, r, k; int pos, key; }p[MAXN];
inline void update(int& p, int l, int r, int x, int v) {
	if (!p) p = ++tot;
	num[p] += v;
	if (l == r) return;
	int mid = (l + r) >> 1;
	if (x <= mid) update(lc[p], l, mid, x, v);
	else update(rc[p], mid + 1, r, x, v);
}
inline void HJTupdate(int x, int v) {
	int k = lower_bound(b + 1, b + len + 1, a[x]) - b;
	for (rg int i = x; i <= n; i += lowbit(i)) update(rt[i], 1, len, k, v);
}
inline int query(int l, int r, int k) {
	if (l == r) return l;
	int mid = (l + r) >> 1, sum = 0;
	for (rg int i = 1; i <= cnt[1]; ++i) sum += num[lc[tmp[1][i]]];
	for (rg int i = 1; i <= cnt[0]; ++i) sum -= num[lc[tmp[0][i]]];
	if (k <= sum) {
		for (rg int i = 1; i <= cnt[1]; ++i) tmp[1][i] = lc[tmp[1][i]];
		for (rg int i = 1; i <= cnt[0]; ++i) tmp[0][i] = lc[tmp[0][i]];
		return query(l, mid, k);
	} else {
		for (rg int i = 1; i <= cnt[1]; ++i) tmp[1][i] = rc[tmp[1][i]];
		for (rg int i = 1; i <= cnt[0]; ++i) tmp[0][i] = rc[tmp[0][i]];
		return query(mid + 1, r, k - sum);
	}
}
inline int HJTquery(int l, int r, int k) {
	memset(tmp, 0, sizeof tmp), cnt[0] = cnt[1] = 0;
	for (rg int i = r; i; i -= lowbit(i)) tmp[1][++cnt[1]] = rt[i];
	for (rg int i = l; i; i -= lowbit(i)) tmp[0][++cnt[0]] = rt[i];
	return query(1, len, k);
}
int main() {
	n = read(), m = read();
	for (rg int i = 1; i <= n; ++i) a[i] = read(), b[++len] = a[i];
	for (rg int i = 1; i <= m; ++i) {
		char s[5]; scanf("%s", s);
		p[i].f = (s[0] == 'Q');
		if (p[i].f)
			p[i].l = read(), p[i].r = read(), p[i].k = read();
		else
			p[i].pos = read(), p[i].key = read(), b[++len] = p[i].key;
	}
	sort(b + 1, b + len + 1);
	len = unique(b + 1, b + len + 1) - b - 1;
	for (rg int i = 1; i <= n; ++i) HJTupdate(i, 1);
	for (rg int i = 1; i <= m; ++i) {
		if (p[i].f)
			printf("%d\n", b[HJTquery(p[i].l - 1, p[i].r, p[i].k)]);
		else {
			HJTupdate(p[i].pos, -1);
			a[p[i].pos] = p[i].key;
			HJTupdate(p[i].pos, 1);
		}
	}
	return 0;
}

整体二分的话就把修改看做减去开始的再加上后来的,跑得比树套树快远了~
整体二分参考代码:

#include <algorithm>
#include <cstdio>
#define rg register
#define file(x) freopen(x".in", "r", stdin), freopen(x".out", "w", stdout)
using namespace std;
template < class T > inline void read(T& s) {
	s = 0; int f = 0; char c = getchar();
	while ('0' > c || c > '9') f |= c == '-', c = getchar();
	while ('0' <= c && c <= '9') s = s * 10 + c - 48, c = getchar();
	s = f ? -s : s;
}

const int _ = 2e5 + 5, INF = 1e9;

int n, m, q, a[_], tr[_], res[_];
int num; struct node { int opt, id, l, r, k; } t[_ << 1], tt1[_ << 1], tt2[_ << 1];

inline void update(int x, int v) { for (rg int i = x; i <= n; i += i & -i) tr[i] += v; }

inline int query(int x) { int res = 0; for (rg int i = x; i >= 1; i -= i & -i) res += tr[i]; return res; }

inline void solve(int ql, int qr, int l, int r) {
	if (ql > qr || l > r) return ;
	if (l == r) { for (rg int i = ql; i <= qr; ++i) if (t[i].opt == 1) res[t[i].id] = l; return ; }
	int mid = (l + r) >> 1, p1 = 0, p2 = 0;
	for (rg int i = ql; i <= qr; ++i) {
		if (t[i].opt == 0) {
			if (t[i].l <= mid) update(t[i].r, t[i].k), tt1[++p1] = t[i]; else tt2[++p2] = t[i];
		} else {
			int cnt = query(t[i].r) - query(t[i].l - 1);
			if (cnt >= t[i].k) tt1[++p1] = t[i]; else t[i].k -= cnt, tt2[++p2] = t[i];
		}
	}
	for (rg int i = 1; i <= p1; ++i) if (tt1[i].opt == 0) update(tt1[i].r, -tt1[i].k);
	for (rg int i = 1; i <= p1; ++i) t[ql + i - 1] = tt1[i];
	for (rg int i = 1; i <= p2; ++i) t[ql + p1 + i - 1] = tt2[i];
	solve(ql, ql + p1 - 1, l, mid), solve(ql + p1, qr, mid + 1, r);
}

int main() {
	read(n), read(m);
	for (rg int i = 1; i <= n; ++i) read(a[i]), t[++num] = (node) { 0, 0, a[i], i, 1 };
	char s[5];
	for (rg int l, r, k, i = 1; i <= m; ++i) {
		scanf("%s", s);
		if (s[0] == 'Q') read(l), read(r), read(k), t[++num] = (node) { 1, ++q, l, r, k };
		else read(l), read(r), t[++num] = (node) { 0, 0, a[l], l, -1 }, t[++num] = (node) { 0, 0, a[l] = r, l, 1 };
	}
	solve(1, num, -INF, INF);
	for (rg int i = 1; i <= q; ++i) printf("%d\n", res[i]);
	return 0;
}
posted @ 2020-01-23 23:39  Sangber  阅读(139)  评论(0编辑  收藏  举报