题目:
思路:
分为两种情况
1.取出一个含 x 数列和一个不含 x 数列,会对答案产生 1 的贡献;
2.取出两个含x数列和哟个不含x的数列,也会贡献1.
然后考虑如何求出每个数出现的次数。
记 st【x】 表示数字 x 上一次出现的位置。初始存在为 0,没出现过为 −1。每次 x 要被改成 y 时,cnt[x] 要加上 i-lst[x],i 为当前的位置,表示这一段中 x 持续存在。然后把 lst【x】 改回 −1,将 y 设为 i。注意所有操作完成后要扫一遍所有数,如果 lst[i] 不为 −1,还要加上最后一段的长度。
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 2e5;
int a[N];
int cnt[N<<1],lst[N<<1];
void solve()
{
int n,m;
cin>>n>>m;
for (int i=1;i<=n+m;i++){
cnt[i]=0,lst[i]=-1;
}
for (int i=1;i<=n;i++){
cin>>a[i];
lst[a[i]]=0;
}
for (int i=1;i<=m;i++){
int p,y;
cin>>p>>y;
if(a[p]!=y) {
cnt[a[p]]+=i-lst[a[p]],lst[a[p]]=-1;
lst[y]=i,a[p]=y;
}
}
for (int i=1;i<=n+m;i++) {
if(lst[i]!=-1) cnt[i]+=m-lst[i]+1;
}
int ans=0;
for (int i=1;i<=n+m;i++){
ans+=cnt[i]*(m+1-cnt[i]);
ans+=cnt[i]*(cnt[i]-1)/2;
}
cout<<ans<<endl;
}
signed main()
{
int t;
cin>>t;
while(t--){
solve();
}
}