P1903 [国家集训队] 数颜色 / 维护队列

带修改的莫队

带修改的莫队就是在基础莫队的基础上增加了一维属性，之前只需要维护l，r现在还需要维护一下时间t，排序还是先按照左端点块儿号排序，然后右端点块儿号排序，最后按时间排序。其它的都是差不多的。

cpp 复制代码

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
//#define x first
//#define y second
//#define int long long
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<int, string> pis;
const int mod = 1e9 + 7;
const int N = 1e6+ 10;
int dx[] = {-1, 0, 1, 0, -1, 1, 1, -1};
int dy[] = {0, 1, 0, -1, 1, 1, -1, -1};
int n, m, mc, mq, len;
int o[N], f[N], st[N], res;
//        结果  标记 
struct query{ // 记录询问的列表
	int l, r, id, t;
}q[N];
struct modify{ // 记录修改操作的列表
	int x, y;
}c[N];

inline int get(int a) // 得到块儿号
{
	return a / len;
}

inline void add(int a) // 增加
{
	if(!st[a]) res ++;
	st[a] ++;
}

inline void del(int a) // 删除
{
	st[a] --;
	if(!st[a]) res --;
}
bool cmp(query a, query b) // 排序
{
	int ai = get(a.l), aj = get(a.r);
	int bi = get(b.l), bj = get(b.r);
	if(ai != bi) return ai < bi; // 按左端点块儿号
	if(aj != bj) return aj < bj; // 按右端点块儿号
	return a.t < b.t; // 按时间
}

inline void sovle()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i ++) cin >> o[i];
	
	for(int i = 0; i < m; i ++)
	{
		char op;
		int a, b;
		cin >> op >> a >> b;
		if(op == 'Q') mq ++, q[mq] = {a, b, mq, mc};
		else c[++ mc] = {a,  b};
	}
	len = pow(n, 0.666); // 怎么分块儿，，，可以找一些大手子的博客看一下
	stable_sort(q + 1, q + mq + 1, cmp);
	int now = 0, l = 1, r = 0;
	for(int i = 1; i <= mq; i ++)
	{
		int id = q[i].id, t = q[i].t;
		while(r < q[i].r) add(o[++ r]);
		while(r > q[i].r) del(o[r --]); // 更新右端点
		while(l < q[i].l) del(o[l ++]);
		while(l > q[i].l) add(o[-- l]); // 更新左端点
		while(now < t) // 更新时间
		{
			now ++;
			if(c[now].x <= r && c[now].x >= l) // 不在修改范围内，直接跳过
			{
				del(o[c[now].x]);
				add(c[now].y);
			}
			swap(o[c[now].x], c[now].y); // 交换两个颜色
		} 
		while(now > t)
		{
			if(c[now].x <= r && c[now].x >= l)
			{
				del(o[c[now].x]);
				add(c[now].y);
			}
			swap(o[c[now].x], c[now].y);
			now --;
		}
		f[id] = res; //  记录结果
	}
	
	for(int i = 1; i <= mq; i ++)
	{
		cout << f[i] << endl;
	}
}

signed main(void)
{
	IOS;
	int t = 1;
//	cin >> t;
	while(t --) sovle();
	return 0;
}