题目链接:https://cn.vjudge.net/problem/ZOJ-4101
题意:有多少中方式,交换两个位置的数字,使两个公式都成立
题解:按照原序列 求出 sum1+=(ll)i*a[i]; sum2+=(ll)i*a[i]*a[i]; 假设i ,j 交换,解方程的到a[i]+a[j]=(y-sum2)/(x-sum1),若x=sum1,y=sum2,那就按照数字一样的算,如果有一个相等或者 (y-sum2) % (x-sum1)!=0,那结果就为0,否则就枚举判断即可,注意不要 / 0,越界等条件
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <vector>
#include <stack>
#include <set>
#include <cmath>
#include <string>
#define ll long long
using namespace std;
const int N=1e5+10;
ll x,y;
int n;
int a[N];
int num[N];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
ll sum1=0,sum2=0;
scanf("%d%lld%lld",&n,&x,&y);
for(int i=1;i<=100000;i++) num[i]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum1+=(ll)i*a[i];
sum2+=(ll)i*a[i]*a[i];
num[a[i]]++;
}
ll ans=0;
if(x==sum1 && y==sum2)
{
for(int i=1;i<=100000;i++)
{
ans+=(ll)(num[i]-1)*num[i]/2;
}
printf("%lld\n",ans);
continue;
}
if(x==sum1 || y==sum2)
{
printf("0\n");
continue;
}
if((y-sum2)%(x-sum1)!=0)
{
printf("0\n");
continue;
}
ll k=(y-sum2)/(x-sum1);
ll cnt;
for(int i=1;i<=n;i++)
{
ll aj=k-a[i],s=a[i]-aj;
// cout<<aj<<" ";
if(s!=0 && k-a[i]>0 && (x-sum1) % s == 0)
{
cnt=(x-sum1)/s;
// cout<<cnt;
cnt+=i;
if(cnt<=n && cnt>i &&a[cnt]==aj) ans++;
}
// cout<<endl;
}
printf("%lld\n",ans);
}
return 0;
}