【CodeVS 5032】【省队集训2016 Day5 T1】Play with array

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 100000;
const int B = 316;
const int BB = 632;

int in() {
	int k = 0, fh = 1; char c = getchar();
	for(; c < ‘0‘ || c > ‘9‘; c = getchar())
		if (c == ‘-‘) fh = -1;
	for(; c >= ‘0‘ && c <= ‘9‘; c = getchar())
		k = (k << 3) + (k << 1) + c - ‘0‘;
	return k * fh;
}

struct BLOCK {
	BLOCK() {
		nxt = NULL;
		len = 0;
		memset(times, 0, sizeof(times));
	}
	BLOCK *nxt;
	int a[BB + B + 3], times[N + 1], len;
};

int cnt = 0;
int n;

namespace BlockList {
	BLOCK *head;
	void Build(BLOCK * t) {head = t;}
	void split(BLOCK *r) {
		int rlen = r->len / 2, tlen = r->len - rlen, to = rlen;
		BLOCK *t = new BLOCK;
		memcpy(t->a + 1, r->a + rlen + 1, sizeof(int) * tlen);
		for(int i = 1; i <= tlen; ++i) {
			++t->times[t->a[i]];
			--r->times[t->a[i]];
		}
		t->len = tlen;
		r->len = rlen;
		t->nxt = r->nxt;
		r->nxt = t;
	}
	void merge(BLOCK *r) {
		BLOCK *t = r->nxt;
		if (t == NULL) return;
		int tlen = t->len, to = r->len;
		memcpy(r->a + to + 1, t->a + 1, sizeof(int) * tlen);
		for(int i = 1; i <= tlen; ++i)
			++r->times[t->a[i]];
		r->len += tlen;
		t = t->nxt;
		delete r->nxt;
		r->nxt = t;
		if (r->len >= BB) split(r);
	}
	BLOCK *find(int &k) {
		BLOCK *t = head;
		while (k - t->len > 0 && t != NULL) {
			k -= t->len;
			t = t->nxt;
		}
		return t;
	}
	void work(int l, int r) {
		BLOCK *t = find(r);
		int num = t->a[r];
		--t->times[num];
		int tlen = --t->len;
		for(int i = r; i <= tlen; ++i)
			t->a[i] = t->a[i + 1];
		if (t->len < B) merge(t);
		t = find(l);
		for(int i = ++t->len; i > l; --i)
			t->a[i] = t->a[i - 1];
		t->a[l] = num;
		++t->times[num];
		if (t->len >= BB) split(t);
	}
	int query(int l, int r, int k) {
		BLOCK *tl = find(l), *tr = find(r);
		int ret = 0;
		if (tl == tr) {
			for(int i = l; i <= r; ++i)
				if (tl->a[i] == k) ++ret;
			return ret;
		} else {
			int lentl = tl->len;
			for(int i = l; i <= lentl; ++i)
				if (tl->a[i] == k) ++ret;
			for(int i = 1; i <= r; ++i)
				if (tr->a[i] == k) ++ret;
			for(tl = tl->nxt; tl != tr && tl != NULL; ret += tl->times[k], tl = tl->nxt);
			return ret;
		}
	}
}

int main() {
	n = in();
	int c = 0, k;
	BLOCK *tmp = new BLOCK;
	BlockList::Build(tmp);
	for(int i = 1; i <= n; ++i) {
		++c;
		if (c > B) {
			c = 1;
			tmp->len = B;
			tmp->nxt = new BLOCK;
			tmp = tmp->nxt;
		}
		k = in();
		tmp->a[c] = k;
		++tmp->times[k];
	}
	tmp->len = c;
	int m = in(), op, l, r;
	while (m--) {
		op = in(); l = in(); r = in();
		if (op == 1)
			BlockList::work(l, r);
		else {
			k = in();
			printf("%d\n", BlockList::query(l, r, k));
		}
	}
	return 0;
}