题目
给你一个长度为n的数组和m次修改,每次修改会将其中的一个数字修改成另一个数字然后形成一个新数组,初始数组和修改过程的数组保证没有相同的数字,问你所有任意两个数组之间去重之后元素的数量的和。
思路
考虑一下所有数组都不含相同元素的情况,那么就有(1+2+3+…+m-2+m-1+m)=m(m+1)/2对,然后每一对都有2n个,那么res结果为m(m+1)/2。然后考虑如何去重就可以了。
在此我们可以假设一下,如果不改变所有元素,那么每个元素都会出现m+1次,对于每一次修改的元素x和被修改的y,如果接下来x不会被修改的话,那么x会增加m-i+1次(前提为所有数组都没有相同的数字),y会减少m-i+1次,遍历一遍找每个元素重复的次数,然后用次数*(次数-1)/2得到每个元素重复所减少的贡献,然后用res减去这些所有的贡献就是最后的结果。
ps:(他奶奶的不会写这个看了题解也不懂,后来才发现每次修改后面的数字都会增加(如果不被修改),属于是题目又没理解到位了qwq)
代码
void slove( )
{
int t;
sc_int(n),sc_int(m);
map<ll,ll>p;
for(int i =1;i<=n;i++)
{
cin>>s[i];
p[s[i]]=m+1;
}
ll res=0;
for(int i=1;i<=m;i++)
{
int l,y;
cin>>l>>y;
int x=s[l];
p[x]-=m-i+1;
p[y]+=m-i+1;
s[l]=y;
}
for(auto i:p){
res+= i.second*(i.second-1)/2 ;
}
cout<<(ll)n*m*(m+1)-res;
cout<<endl;
}