codeforces C2 - Adjust The Presentation

cf原题链接

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 200010;
int a[N]; // 员工序列
int b[N]; // 幻灯片对应员工编号   每次用一下pos 换成新编号
int pos[N]; // 员工序列重新编号
set<int>id[N];
bool st[N]; // 记录上一次的贡献
int n, m, q;
int sum;
int check(int p)  //检查每个点贡献是否更新 （ 2 到 n
{
	if (p > n || p == 1)return 0;
	int num = *id[p].begin() >= *id[p - 1].begin();
	int res = num - st[p];
	st[p] = num;
	return res;
}

void COUT()
{
	if (sum == n - 1)cout << "YA" << endl;
	else cout << "TIDAK" << endl;
}

void solve()
{

	cin >> n >> m >> q;
	for (int i = 1; i <= n; i++)id[i].clear(), id[i].insert(200010), st[i] = false;
	for (int i = 1; i <= n; i++)cin >> a[i], pos[a[i]] = i;

	for (int i = 1; i <= m; i++)
	{
		cin >> b[i];
		b[i] = pos[b[i]]; // 按照a[i]为顺序重新编号
		id[b[i]].insert(i); // 存下所有a[i]出现的位置
	}

	sum = 0;
	for (int i = 1; i <= n; i++)
		sum += check(i);
	COUT();

	while (q--)
	{
		int l, r;cin >> l >> r;
		r = pos[r];

		id[b[l]].erase(l);  // 更新幻灯片对应员工
		id[r].insert(l);

		sum += check(b[l]) + check(b[l] + 1) + check(r) + check(r + 1); // 每次操作会变化的点只有这一点和这一点后面一点
		COUT();
		
		b[l] = r; // 永久更新
	}
}

signed main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int t = 1;
	cin >> t;
	while (t--) solve();
	return 0;
}

题意：判断幻灯片序列里员工第一次出现的相对位置是否和初始员工位置相同

PS：这里需要注意的就是要防止查询是否相同时超时， 一不小心就会O（qn），优化就是每次记录一下每个成员是否大于上个成员，q次操作每次只需更新变动的成员的贡献