1. 程式人生 > >「SCOI 2014」方伯伯的 OJ「動態開點線段樹」

「SCOI 2014」方伯伯的 OJ「動態開點線段樹」

題目傳送門

備註:洛谷,2018/9/26,此題題面有誤,見討論。

題解

動態開點線段樹。

線段樹的葉子的結點有個值, v v ,它表示這個位置上的人編號是多少。

我們記錄兩個變數 l b ,

r b lb,rb 表示當前位置的最小值和最大值。

每次把一個人提到第一時, 把他從原位置刪除,放到 l b 1

lb-1 ;提到最後也是一樣。

詢問的時候就線上段樹上詢問從左往右第 k k 個位置上的人的編號,因此每個線段樹結點還要維護一個 s u m

sum ,表示這個區間內有多少使用者。

最後還注意一點,由於編號過大,使用 m a p < i n t ,    i n t > map<int,\;int> 儲存每個人線上段樹中的位置.

#include <cstdio>
#include <map>
using namespace std;

#define L(x) max(x, 1)
#define R(x) min(x, n)

const int N = 1e5 + 10;

int n, m, Lb, Rb;
map<int, int> mp;

struct Seg {
	int l, r, v, sum;
} tr[N * 64];

int rt, tot;

int modify(int & k, int l, int r, int pos, int val) {
	if(!k) {
		tr[k = ++ tot].sum = max(0, R(r) - L(l) + 1);
	}
	if(l == r) {
		tr[k].v = val;
		return 0;
	}
	int mid = l + r >> 1;
	int lsum = tr[k].l ? tr[tr[k].l].sum : max(0, R(mid) - L(l) + 1);
	if(pos <= mid) return modify(tr[k].l, l, mid, pos, val);
	return lsum + modify(tr[k].r, mid + 1, r, pos, val);
}

int update(int & k, int l, int r, int pos, int val, int id) {
	if(!k) {
		tr[k = ++ tot].sum = max(0, R(r) - L(l) + 1);
		if(l == r) tr[k].v = l;
	}
	tr[k].sum += val;
	if(l == r) {
		tr[k].v = id;
		return 0;
	}
	int mid = l + r >> 1;
	int lsum = tr[k].l ? tr[tr[k].l].sum : max(0, R(mid) - L(l) + 1);
	if(pos <= mid) return update(tr[k].l, l, mid, pos, val, id);
	return lsum + update(tr[k].r, mid + 1, r, pos, val, id);
}

int query(int & k, int l, int r, int x) {
	if(!k) {
		tr[k = ++ tot].sum = max(0, R(r) - L(l) + 1);
		if(l == r) tr[k].v = l;
	}
	if(l == r) return tr[k].v;
	int mid = l + r >> 1;
	int lsum = tr[k].l ? tr[tr[k].l].sum : max(0, R(mid) - L(l) + 1);
	if(x <= lsum) return query(tr[k].l, l, mid, x);
	return query(tr[k].r, mid + 1, r, x - lsum);
}

int main() {
	scanf("%d%d", &n, &m);
	Lb = 1, Rb = n;
	for(int i = 1, a = 0, opt, x, y; i <= m; i ++) {
		scanf("%d%d", &opt, &x), x -= a;
		if(opt == 1) {
			scanf("%d", &y), y -= a;
			mp[y] = mp.count(x) ? mp[x] : x;
			printf("%d\n", a = 1 + modify(rt, -m, n + m, mp[y], y));
		} else if(opt == 2) {
			printf("%d\n", a = 1 + update(rt, -m, n + m, mp.count(x) ? mp[x] : x, -1, -1));
			update(rt, -m, n + m, mp[x] = -- Lb, 1, x);
		} else if(opt == 3) {
			printf("%d\n", a = 1 + update(rt, -m, n + m, mp.count(x) ? mp[x] : x, -1, -1));
			update(rt, -m, n + m, mp[x] = ++ Rb, 1, x);
		} else {
			printf("%d\n", a = query(rt, -m, n + m, x));
		}
	}
	return 0;
}