【数学+分类讨论】Element Swapping（ZOJ-4101）

链接

ZOJ-4101

题意

$数组a通过交换一对数字,得到了b数组。$
$给出如下的两个值x={\sum }_{k=1}^{n}k{a}_k和y={\sum }_{k=1}^{n}k{a}_k^2和b数组.$
$问可能有多少种交换方式使得满足条件。$
$x和y的值有可能是和数组不匹配的，这样的情况时输出0。$

题解

$已知x={\sum }_{k=1}^{n}k{a}_k和y={\sum }_{k=1}^{n}k{a}_k^2，$
$则令x^{\prime }={\sum }_{k=1}^{n}k{b}_k和y^{\prime }={\sum }_{k=1}^{n}k{b}_k^2，$
$假设a_i，a_j交换成为了b_i，b_j，即a_i=b_j，a_j=b_i。$

$有X=x^{\prime }-x=i{b}_i+j{b}_j-i{a}_i-j{a}_j=(b_i-b_j)\ast (i-j)$
$有Y=y^{\prime }-y=i{b}_i^2+j{b}_j^2-i{a}_i^2-j{a}_j^2=(b_i^2-b_j^2)\ast (i-j)$

$易得，\frac{Y}{X}=\frac{(b_i^2-b_j^2)\ast (i-j)}{(b_i-b_j)\ast (i-j)}=b_i+b_j=C（常数）$
$我们知道了两个交换的数之和，则可通过b_i求得b_j=\frac{Y}{X}-b_i，枚举b_i，可求得b_j,然后可通过X=(b_i-b_j)\ast (i-j)得到j的下标i-\frac{X}{b_i-b_j},只需验证求得的下标j是否满足b_j即可。$

$需要注意的是:$
$当X\ne 0，Y\%X\ne 0时,x,y不匹配，ans=0;$
$当X=0，Y\ne 0时,x,y不匹配，ans=0;$
$当X=Y=0时，因为交换前后无变化，则a_i=a_j,枚举每个数出现的次数nu{m}_i，ans={\sum }_{i=1}^n\frac{nu{m}_i\ast (nu{m}_i-1)}{2}$

代码

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;

#define ll long long

ll T,n,x,y,c;
ll suma,sumb,sum;
ll a[200500],b[200500];
unordered_map<ll,ll>dyy;
int main(){
	cin>>T;
	while(T--)
	{
		cin>>n>>x>>y;
		suma=sumb=sum=0;
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];suma+=a[i]*i;
			b[i]=a[i]*a[i];sumb+=b[i]*i;
		}
		x=suma-x;//X
		y=sumb-y;//Y
		if(x!=0)
		{
			if(y%x)
			{
				cout<<0<<endl;
				continue;
			}
			else c=y/x;//c=ai+aj
			for(int i=1;i<n;i++)
			{
				ll aj=c-a[i];
				if(a[i]-aj==0)continue;//检查除数a_i-a_j是否为0，因为X>0,为0则跳过
				if(x%(a[i]-aj))continue;//防止x和y不匹配
				int j=i-x/(a[i]-aj);//计算j的下标
				if(j<=i||j>n)continue;//防止重复，对于每个i只查找之后的j
				if(a[j]==aj)sum++;
			}
		}
		else
		{
			if(y!=0)
			{
				cout<<0<<endl;
				continue;
			}
			dyy.clear();
			for(int i=1;i<=n;i++)
			{
				dyy[a[i]]++;
			}
			for(auto t:dyy)
			sum+=t.second*(t.second-1)/2;
		}
		cout<<sum<<endl;
	}
	return 0;
}