loj#6487 基础 FFT 练习题

分析

仔细观察a[i],b[i]的数据范围

于是我们转为枚举所有a[i],b[i]的值

然后暴力即可

代码

#include<bits/stdc++.h>
using namespace std;
#define abs(x) x>0?x:-x
int s[],sum1[],sum2[],a[],b[],n,c1,c2;
long long Ans;
inline int ra(){
    int x=;char s=getchar();
    while(!isdigit(s))s=getchar();
    while(isdigit(s))x=(x<<)+(x<<)+(s-''),s=getchar();
    return x;
}
int main(){
    int t,i,j,k;
    for(i=;i<=;++i)s[i*i]=i;
    for(i=;i<=;++i)s[i]=(s[i]?s[i]:s[i-]);
    t=ra();
    while(t--){
      n=ra();
      for(i=;i<=n;++i){
          k=ra();
          sum1[k]?:a[++c1]=k;
          sum1[k]++;
      }
      for(i=;i<=n;++i){
          k=ra();
          sum2[k]?:b[++c2]=k;
          sum2[k]++;
      }
      for(i=;i<=c1;++i){
          for(j=;j<=c2;++j)
            Ans+=1ll*sum1[a[i]]*sum2[b[j]]*s[abs((a[i]-b[j]))];
          sum1[a[i]]=;
      }
      for(i=;i<=c2;++i)sum2[b[i]]=;
      cout<<Ans<<"\n";
      c1=c2=Ans=;
    }
    return ;
}