【jzoj3723】【srm603】【Sum Of Arrays】【fft】

题目大意

有两个长度为n整数数列A 和B。（n<=1e5）把A，B按某种顺序排序生成C，C[i] = A[i] + B[i]。目标是让C 里面产生尽可能多的相同数字，求最多出现的数字出现多少次和这时最大是多少。你会获得整数n 以及用于产生数列A 和B 的随机种子。每一组随机种子有六个整数，用a0, a1,… ,a5 以及b0, b1,…,b5表示。数列A 按照如下定义生成：A[0] = a0。A[1] = a1。对于在2 到n-1之间的i，有A[i] = (A[i-1]* a2 + A[i-2]*a3 + a4)mod a5。同理，数列B 由类似的定义生成。

解题思路

我们可以分类处理，当一个数出现次数大于等于10时，这些数不会超过1e4，我们可以暴力做，同时留下小于10的数。对于小于10的数我们一个一个考虑,A,B取值是0和1，我们可以发现a[i]+b[j]=c[k]=1=>a[i]*b[k-j]=c[k]=1。统计这个数量就是卷积。我们可以用fft优化。

TLE-code

#include<cmath>
#include<cstdio>
#include<algorithm>
#define LF double
#define LL long long
#define min(a,b) ((a<b)?a:b)
#define max(a,b) ((a>b)?a:b)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const mxn=1e5;LF pi=acos(-1);
int n,N,M,a[10],b[10],A[262144+10],B[262144+10],AA[262144+10],BB[262144+10],cnt[262144+10],pos[262144+10];
struct rec{
    LF x,y;
    rec(LF X=0,LF Y=0){x=X,y=Y;}
};
rec operator+(rec x,rec y){return rec(x.x+y.x,x.y+y.y);}
rec operator-(rec x,rec y){return rec(x.x-y.x,x.y-y.y);}
rec operator*(rec x,rec y){return rec(x.x*y.x-x.y*y.y,x.x*y.y+x.y*y.x);}
rec x[262144+10],y[262144+10],t[262144+10];int tmp,k;LF tmp2;
void DFT(rec *a,int tag){
    fo(i,0,N-1){t[pos[i]]=a[i];}
    for(int size=2;size<=N;size=size<<1)
        for(int half=size>>1,i=0;i<half;i++){
            tmp2=tag*pi*i/half;
            rec w=rec(cos(tmp2),sin(tmp2));
            for(int j=i;j<N;j=j+size){
                k=j+half;rec u=t[j],v=w*t[k];
                t[j]=u+v;
                t[k]=u-v;
            }
        }
    fo(i,0,N-1)a[i]=t[i];
}
int main(){
    freopen("sum.in","r",stdin);
    freopen("sum.out","w",stdout);
    int mx1,tmp;
    while(scanf("%d",&n)!=EOF){
        fo(i,0,5)scanf("%d",&a[i]);
        fo(i,0,5)scanf("%d",&b[i]);
        fo(i,0,mxn-1)A[i]=B[i]=0;
        A[a[0]]++;A[a[1]]++;N=max(max(a[0],a[1]),max(b[0],b[1]));
        fo(i,2,n-1){A[a[i&1]=(1ll*a[!(i&1)]*a[2]%a[5]+1ll*a[i&1]*a[3]%a[5]+a[4])%a[5]]++;N=max(N,a[i&1]);}
        B[b[0]]++;B[b[1]]++;
        fo(i,2,n-1){B[b[i&1]=(1ll*b[!(i&1)]*b[2]%b[5]+1ll*b[i&1]*b[3]%b[5]+b[4])%b[5]]++;N=max(N,b[i&1]);}
        AA[0]=0;fo(i,0,N)if(A[i]>=10)AA[++AA[0]]=i;
        BB[0]=0;fo(i,0,N)if(B[i]>=10)BB[++BB[0]]=i;
        fo(i,0,N*2)cnt[i]=0;
        fo(i,1,AA[0])fo(j,1,BB[0])cnt[AA[i]+BB[j]]+=min(A[AA[i]],B[BB[j]])-9;
        mx1=N,M=log(N)/log(2)+2,N=pow(2,M);
        fo(i,0,N-1){
            tmp=0;
            for(int ii=i,j=1;j<=M;tmp=(tmp<<1)+(ii&1),j++,ii=ii>>1);
            pos[i]=tmp;
        }
        fo(now,1,9){
            fo(i,0,N-1)x[i]=rec(A[i]>=now,0),y[i]=rec(B[i]>=now,0);
            DFT(x,1);DFT(y,1);
            fo(i,0,N-1)x[i]=x[i]*y[i];
            DFT(x,-1);
            fo(i,0,mx1*2)cnt[i]=cnt[i]+(x[i].x+0.5)/N;
        }
        int mx=-1e9,num=0;
        fo(i,0,mx1*2)if(cnt[i]>=mx){mx=cnt[i],num=i;}
        printf("%d %d\n",mx,num);
    }
    return 0;
}