The 16th Zhejiang Provincial Collegiate Programming Contest - B Element Swapping（zoj4101）

求出b数组的x，y值，再与原x、y相减得

             dx=(i*b[i]+j*b[j])-(i*b[j]+j*b[i])=(i-j)*(b[i]-b[j])

   同理  dy=(i-j)*(b[i]^2-b[j]^2)=dx*(b[i]+b[j])

  所以直接枚举i 将求出的 b[j]=dy/dx-b[i] 带入上面第一个式子求出 j，判断 b[j]和求出的是否一致。

 同时特判下dx,dy为0时的情况即可。。

 

 

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
long long B[100010];
int cnt[100010];
int main(){
    int t,i,n;
    scanf("%d",&t);
    long long x,y,a,b,k,u;
    while(t--){
        scanf("%d%lld%lld",&n,&x,&y);
        a=b=0;
        memset(cnt,0,sizeof(cnt));
        for(i=1;i<=n;i++){
            scanf("%lld",&B[i]);
            a+=i*B[i];
            b+=i*B[i]*B[i];
        }
        x=a-x;
        y=b-y;
        long long ans=0;
        if(x==0){
            if(y==0){
                for(i=1;i<=n;i++){
                    ans+=cnt[B[i]];
                    cnt[B[i]]++;
                }
            }
        }
        else{
            if(y%x==0){
                y/=x;
                for(i=1;i<=n;i++){
                    k=B[i]-(y-B[i]);
                    if(k&&x%k==0){
                        u=i-x/k;
                        if(u>=1&&u<i&&B[u]==y-B[i]) ans++;
                    }
                }
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}