2019浙江省赛B zoj4101 Element Swapping(推公式)

传送门

题意：数组a通过交换一对数字,得到了b数组,给出x=∑nk=1kak和y=∑nk=1ka2k和b数组,问有多少对l,r(l<=r)能满足条件
 

思路：

假设交换了 (i,j)，那么：i*a[i] -> i*a[j]，j*a[j] -> j*a[i]

因此对于原来的 x、y 有：

可以得到：

将得到的两个式子相除，有：

第一个式子可以化简为 X−x=(ai−aj)∗(j−i)

• 第一个式子左边为一个定值,所以ai和aj一一对应,可以枚举a数组
• 第二个数组可以化为X2−X1+(bj−bi)ibj−bi=j,假设j>i(方便计数),枚举ai通过式子一可以计算aj,然后通过式子二可以计算出j,判一下aj是否等于j
• X2−X1=0,Y2−Y1≠0,则不合法
• X2−X1=0,Y2−Y1=0推出ai=aj,j≠i
   

  #include<iostream>
  #include<algorithm>
  #include<cstring>
  #include<cstdio>
  #include<map>
  #define ll long long
  #define N 0x3f3f3f3f
  using namespace std;
  ll a[100010];
  int main()
  {
  	int t;
  	scanf("%d",&t);
  	while(t--)
  	{
  		int n;
  		map<ll,ll>mp;
  		ll x,y;
  		scanf("%d%lld%lld",&n,&x,&y);
  		ll x1=0,y1=0;
  		ll minn=N;
  		for(int i=1;i<=n;i++)
  		{
  			scanf("%lld",&a[i]);
  			mp[a[i]]++;
  			x1=x1+i*a[i];
  			y1=y1+i*a[i]*a[i];
  		}
  		ll A=x1-x,B=y1-y;;
  		if(A==0)
  		{
  			if(B!=0)
  			{
  				cout<<0<<endl;
  				continue;
  			}
  			ll sum=0;
  			map<ll,ll>::iterator it;
  			for(it=mp.begin();it!=mp.end();it++)
  			{
  				sum+=((it->second)-1)*(it->second)/2;
  			}
  			printf("%lld\n",sum);
  			continue;
  		}
  		if(B%A)
  		{
  			printf("0\n");
  			continue;
  		}
  		ll dt=B/A,sum=0;
  		for(int i=1;i<=n;i++)
  		{
  			ll aj=dt-a[i];
  			if(aj==a[i])
  			continue;
  			ll j=(A+(aj-a[i])*i)/(aj-a[i]);
  			if(j<=i||j>n)
  			continue;
  			if(a[j]==aj)
  			sum++;
  		}
  		printf("%lld\n",sum);
  	}
  	return 0;
  }

  不知道为啥 数组a 我定义成int类型的就wa 0.0 明明数组范围是 10^6