AIZU 2560 [想法题]

表示敲完多项式乘法&高精度乘法两道FFT模板题后就开始来磕这题了

这题相对而言应该不算模板题 不过神犇们肯定还是一眼看穿

如果原OJ访问速度较慢的话可以考虑戳这里

http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=153505

-------------------------------------------------------------------------------------------------------------

构造什么的 还是太巧妙了

比如对于这样的样例

2

1 2

3 4

可以构造出这样的矩阵（原矩阵在左上角 额外的矩阵在右下角 且两矩阵不相交且中心对称）

a矩阵               b矩阵

1 2 0 0           0 0 0 0

3 4 0 0           0 0 0 0

0 0 0 0           0 0 4 3

0 0 0 0           0 0 2 1

如果变为多项式的话 就是这样

多项式a  1 2 0 0 3 4 0 0 0 0 0 0 0 0 0 0

多项式b  0 0 0 0 0 0 0 0 0 0 4 3 0 0 2 1

当我们把它们乘起来后 会发现贡献到同一项点对的距离是一样的

当n不为2的整数幂的时候 可以像fft模板题一样 通过补零来进行构造

构造出这两个多项式后 我们就可以直接套fft模板 然后对结果多项式的有效的位数都访问一次即可统计出答案

-------------------------------------------------------------------------------------------------------------

不过A掉后还是有一个问题还没弄懂

设多项式a，b的长度为n 那么按照常规的多项式乘法 我们应该把他们后面都补上n个0

因为结果多项式的长度为n×2 （虽然实际有效位数比n还小）

然而我参考的某AC代码 并没有用n×2的长度 而是只用到n的长度让结果 “自然溢出”了 （不是标准说法）

而且貌似n位之后的部分溢出到前面几位了

如果有神犇知道是为什么的话希望能回复下

#include <bits/stdc++.h>
using namespace std;
const int N=<<,M=<<;
const double pi=acos(-);
typedef complex <double> E;
int mp[M][M];
int n,m,L;
int R[N];
long long cnt[N];
E a[N],b[N];
int sum;
double ans;
void fft(E *A,int f)
{
    for(int i=;i<n;++i)
        if(i<R[i])
            swap(A[i],A[R[i]]);
    for(int i=;i<n;i<<=)
    {
        E wn(cos(pi/i),sin(pi/i)*f);
        for(int p=i<<,j=;j<n;j+=p)
        {
            E w(,);
            for(int k=;k<i;++k,w*=wn)
            {
                E x=A[j+k],y=w*A[j+k+i];
                A[j+k]=x+y;
                A[j+k+i]=x-y;
            }
        }
    }
}
int getdist(int x,int y)
{
    return x*x+y*y;
}
int main()
{
    scanf("%d",&m);
    for(n=;n<m*;n<<=)
        ++L;
    for(int i=;i<m;++i)
        for(int j=;j<m;++j)
        {
            scanf("%d",&mp[i][j]);
            sum+=mp[i][j];
            cnt[]-=mp[i][j];
            a[i*n+j]=mp[i][j];
            b[(n-i-)*n+(n-j-)]=mp[i][j];
        }
    n=n*n*;
    L=L*+;
    for(int i=;i<n;++i)
        R[i]=(R[i>>]>>)|((i&)<<(L-));
    fft(a,);
    fft(b,);
    for(int i=;i<n;++i)
        a[i]*=b[i];
    fft(a,-);
    int tn=sqrt(n/)/,x,y;
    x=tn;
    y=tn;
    for(int i=*tn*tn-*tn-;i;--i)
    {
        int tmp1;
        if(y>=tn)
            tmp1=getdist(x-(tn*-),y-(tn*-));
        else
            tmp1=getdist(x--(tn*-),tn*--(tn*--y-));
        double tmp2=a[x*tn*+y].real()/n;
        ans+=sqrt(tmp1)*tmp2;
        cnt[tmp1]+=tmp2+0.5;
        x+=(y==tn*-);
        y=(y==tn*-?:y+);
    }
    printf("%.10f\n",ans/((long long)sum*(sum-)));
    int flag=;
    for(int i=;i<m*m*;++i)
        if(cnt[i])
        {
            printf("%d %lld\n",i,cnt[i]/);
            if(++flag>=)
                break;
        }
    return ;
}

还是把参考的那个只用了n的长度的多项式的神犇的代码贴一下（如果原作者有意见的话我会删掉的……）

#include<stdio.h>
#include<algorithm>
#include<complex>
#include<vector>
#include<math.h>
#include<map>
using namespace std;
const double PI = 4.0*atan(1.0);
typedef complex<double> Complex;
const Complex I(, );
void fft(int n, double theta, Complex a[]) {
  for (int m = n; m >= ; m >>= ) {
    int mh = m >> ;
    for (int i = ; i < mh; i++) {
      Complex w = exp(i*theta*I);
      for (int j = i; j < n; j += m) {
        int k = j + mh;
        Complex x = a[j] - a[k];
        a[j] += a[k];
        a[k] = w * x;
      }
    }
    theta *= ;
  }
  int i = ;
  for (int j = ; j < n - ; j++) {
    for (int k = n >> ; k > (i ^= k); k >>= );
    if (j < i) swap(a[i], a[j]);
  }
}
int b[][];
Complex p[<<];
Complex q[<<];
int main(){
    int a;
    scanf("%d",&a);
    int cnt=;
    for(int i=;i<a;i++){
        for(int j=;j<a;j++)scanf("%d",&b[i][j]);
    }
    int n=;
    while(n<a*)n*=;
    for(int i=;i<a;i++)for(int j=;j<a;j++){
        cnt+=b[i][j];
        p[i*n+j]=q[(n--i)*n+(n--j)]=Complex(b[i][j],);
    }
    fft(n*n,*PI/n/n,p);
    fft(n*n,*PI/n/n,q);

    for(int i=;i<n*n;i++)p[i]=p[i]*q[i];
    fft(n*n,-*PI/n/n,p);
    for(int i=;i<n*n;i++)p[i]/=n*n;
    /*for(int i=0;i<n;i++){
        for(int j=0;j<n;j++)printf("%.0f ",p[i*n+j].real());
        printf("\n");
    }*/
    double ret=;
    long long sz=;
    map<int,long long>m;
    for(int i=;i<n*n;i++){
            int pi=i/n;
            int pj=i%n;
            if(pj>=n/)continue;
            if(pj==&&pi>=n/)continue;
            long long v=(long long)(p[n*n--i].real()+0.5);
            if(i==){v-=cnt;v/=;}
            if(!v)continue;

            int I=min(pi,n-pi);
            int J=pj;
            int t=I*I+J*J;
            if(m.count(t)){
                m[t]+=v;
            }else{
                m[t]=v;
            }
            sz+=v;
            ret+=sqrt(t)*v;
    }
    printf("%.12f\n",ret/sz);
    int at=;
    for(map<int,long long>::iterator it=m.begin();it!=m.end();it++){
        if(at==)break;
        at++;
        printf("%d %lld\n",(*it).first,(*it).second);
    }
}