C. Serval and Toxel‘s Arrays（容斥+按值算贡献

Problem - C - Codeforces

题意：一个 n 长的值成对不同数组，有 m 次变化，每次执行一次 p 位置，改为 v 值【保证每次改完的数组内的值成对不同】，计算所有 m + 1 的数组两两去重合并之后不同的数的个数的和；

分析：由于确保两两必然不同，很好想到这题的关键在于值而非位置，对于每个值算贡献 —— 因为同样的值，产生贡献的情况要不是 1 ，要不是 2，对于合并后数组内有两个相同值的贡献很好算 —— 就是,但是 1 的情况实在不好想，想了一段时间突然发现确实难算，后来想到容斥一下就行了 —— 总数 m * （m - 1） / 2 * n * 2；

所以最终就是 总数减去重复计算的即可

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 15;
const int mo = 998244353;
#define pb push_back
#define pii pair<int,int>
#define ft first
#define sd second
#define ffor(i,a,b,c) for(int i=(a);i<(b);i+=(c))
#define For(i,a,b,c) for(int i=(a);i<=(b);i+=(c))
#define rfor(i,a,b,c) for(int i=(a);i>(b);i-=(c))
#define Rfor(i,a,b,c) for(int i=(a);i>=(b);i-=(c))
#define all(x) (x).begin(), (x).end()
#define debug1(x) cerr<<"! "<<x<<endl;
#define debug2(x,y) cerr<<"#  "<<x<<" "<<y<<endl;

void slv() {
	int n, m; cin >> n >> m;
	vector<int>a(n + 1);
	vector<int> nm(n + m + 1), pr(n + m + 1, -1);
	For(i, 1, n, 1) {
		cin >> a[i];
		nm[a[i]]++;
		pr[a[i]] = 1;
	}
	For(i, 2, m + 1, 1) {
		int p, v; cin >> p >> v;
		int u = a[p];
		if (v == u) continue;
		nm[v] += 1;
		pr[v] = i;
		a[p] = v;
		nm[u] += (i - pr[u] - 1);
		pr[u] = -1;
	}
	For(i, 1, n + m, 1) {
		if (pr[i] != -1) {
			nm[i] += m + 1 - pr[i];
		}
	}
	ll tot = 1ll * (m + 1) * m * n;
	For(i, 1, n + m, 1) {
		if (nm[i]) {
			tot -= 1ll * (nm[i] - 1) * nm[i] / 2;
		}
	}
	cout << tot << '\n';
}
int main() {
	ios::sync_with_stdio(false); cin.tie(0);
	int t; cin >> t;
	while (t--) {
		slv();
	}
}