CF1789C Serval and Toxel‘s Arrays

题目:

 思路:  

分为两种情况

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();
	}
	
	
}

  • 8
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值