bzoj 4836 [Lydsy1704月赛]二元运算分治FFT+生成函数

[Lydsy1704月赛]二元运算

Time Limit: 8 Sec Memory Limit: 128 MB
Submit: 577 Solved: 201
[Submit][Status][Discuss]

Description

定义二元运算 opt 满足

现在给定一个长为 n 的数列 a 和一个长为 m 的数列 b ，接下来有 q 次询问。每次询问给定一个数字 c

你需要求出有多少对 (i, j) 使得 a_i opt b_j=c 。

Input

第一行是一个整数 T (1≤T≤10) ，表示测试数据的组数。

对于每组测试数据：

第一行是三个整数 n,m,q (1≤n,m,q≤50000) 。

第二行是 n 个整数，表示 a_1,a_2,?,a_n (0≤a_1,a_2,?,a_n≤50000) 。

第三行是 m 个整数，表示 b_1,b_2,?,b_m (0≤b_1,b_2,?,b_m≤50000) 。

第四行是 q 个整数，第 i 个整数 c_i (0≤c_i≤100000) 表示第 i 次查询的数。

Output

对于每次查询，输出一行，包含一个整数，表示满足条件的 (i, j) 对的个数。

Sample Input

2
2 1 5
1 3
2
1 2 3 4 5
2 2 5
1 3
2 4
1 2 3 4 5

Sample Output

1
0
1
0
0
1
0
1
0
1

HINT

这道题就是分治fft的生成函数题，加法应该很好办，但是减法怎么办，就是将其变成加法就可以了。

减法的时候，对于x-y，变成x-mid-1,mid-y，这样就相加就变成了x-y-1,最后把1加上去就可以了。

 #include<cstring>

 #include<cmath>

 #include<algorithm>

 #include<cstdio>

 #include<iostream>

 #define pi acos(-1)

 #define ll long long

 #define N 100007

 using namespace std;

 inline int read()

 {

     int x=,f=;char ch=getchar();

     while(!isdigit(ch)){if(ch=='-')f=-;ch=getchar();}

     while(isdigit(ch)){x=(x<<)+(x<<)+ch-'';ch=getchar();}

     return x*f;

 }

 int n,m,q,mx;

 int a[N],b[N],rev[N<<];

 ll ans[N];

 struct comp

 {

     double r,v;

     comp(){r=v=0.0;}

     comp(double x,double y){r=x,v=y;}

     void init(){r=v=0.0;}

     friend inline comp operator+(comp x,comp y){return comp(x.r+y.r,x.v+y.v);}

     friend inline comp operator-(comp x,comp y){return comp(x.r-y.r,x.v-y.v);}

     friend inline comp operator*(comp x,comp y){return comp(x.r*y.r-x.v*y.v,x.r*y.v+x.v*y.r);}

     friend inline comp operator/(comp x,int y){return comp(x.r/y,x.v/y);}

 }c[N<<],d[N<<];

 void FFT(comp *a,int flag,int num)

 {

     for (int i=;i<num;i++)

         if (i<rev[i]) swap(a[i],a[rev[i]]);

     for (int i=;i<num;i<<=)

     {

         comp wn=comp(cos(pi/i),flag*sin(pi/i));

         for (int j=;j<num;j+=(i<<))

         {

             comp w=comp(,);

             for (int k=;k<i;k++,w=w*wn)

             {

                 comp x=a[j+k],y=w*a[i+j+k];

                 a[j+k]=x+y,a[j+k+i]=x-y;

             }

         }

     }

     if (flag==-) for (int i=;i<num;i++) a[i].r=a[i].r/num;

 }

 void CDQ(int l,int r)

 {

     if (l==r)

     {

         ans[]+=(ll)a[l]*b[l];

         return;

     }

     int mid=(l+r)>>,num,up=r-l+,L=;

     for (num=;num<=up;num<<=,L++);if (L) L--;

     for (int i=;i<num;i++) rev[i]=(rev[i>>]>>)|((i&)<<L);

     memset(c,,sizeof(c[])*num),memset(d,,sizeof(d[])*num);

     for (int i=l;i<=mid;i++) c[i-l].r=a[i];

     for (int i=mid+;i<=r;i++) d[i-mid-].r=b[i];

     FFT(c,,num),FFT(d,,num);

     for (int i=;i<num;i++) c[i]=c[i]*d[i];

     FFT(c,-,num);

     for (int i=;i<num;i++) ans[i+l+mid+]+=(ll)(c[i].r+0.5);

     memset(c,,sizeof(c[])*num),memset(d,,sizeof(d[])*num);

     for (int i=mid+;i<=r;i++) c[i-mid-].r=a[i];

     for (int i=l;i<=mid;i++) d[mid-i].r=b[i];

     FFT(c,,num),FFT(d,,num);

     for (int i=;i<num;i++) c[i]=c[i]*d[i];

     FFT(c,-,num);

     for (int i=;i<num;i++) ans[i+]+=(ll)(c[i].r+0.5);

     CDQ(l,mid),CDQ(mid+,r);

 }

 int main()

 {

     int T=read();

     while(T--)

     {

         n=read(),m=read(),q=read();

         memset(a,,sizeof(a)),memset(b,,sizeof(b)),memset(ans,,sizeof(ans));

         for (int i=;i<=n;i++)

         {

             int x=read();

             a[x]++,mx=max(mx,x);

         }

         for (int i=;i<=m;i++)

         {

             int x=read();

             b[x]++,mx=max(mx,x);

         }

         CDQ(,mx);

         for (int i=;i<=q;i++)

             printf("%lld\n",ans[read()]);

     }

 }

#include<cstring>#include<cmath>#include<algorithm>#include<iostream>#include<cstdio>
#define pi acos(-1)#define N 600007using namespace std;inline int read(){int x=0,f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}
int R,C,H,W,num,L;int rev[N],a[507][507],b[507][507];struct comp{double r,v;comp(){r=v=0.0;}comp(double x,double y){r=x,v=y;}void init(){r=v=0.0;}friend inline comp operator+(comp x,comp y){return comp(x.r+y.r,x.v+y.v);}friend inline comp operator-(comp x,comp y){return comp(x.r-y.r,x.v-y.v);}friend inline comp operator*(comp x,comp y){return comp(x.r*y.r-x.v*y.v,x.r*y.v+x.v*y.r);}friend inline comp operator/(comp x,int y){return comp(x.r/y,x.v/y);}}a1[N],b1[N],a2[N],b2[N],c[N];char ch[507],T[507][507];
void FFT(comp *a,int flag){for (int i=0;i<num;i++)if (i<rev[i]) swap(a[i],a[rev[i]]);for (int i=1;i<num;i<<=1){comp wn=comp(cos(pi/i),flag*sin(pi/i));for (int j=0;j<num;j+=(i<<1)){comp w=comp(1,0);for (int k=0;k<i;k++,w=w*wn){comp x=a[j+k],y=w*a[j+k+i];a[j+k]=x+y,a[j+k+i]=x-y;}}}if (flag==-1) for (int i=0;i<num;i++) a[i].r=a[i].r/num;}int main(){R=read(),C=read();for (int i=0;i<R;i++){scanf("%s",ch);for (int j=0;j<C;j++){if (ch[j]=='G') a1[i*C+j]=comp(1,0);else b1[i*C+j]=comp(1,0);}}for (num=1;num<=R*C*2;num<<=1,L++);if (L) L--;for (int i=0;i<num;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<L);FFT(a1,1),FFT(b1,1);int Total=read();for (int Case=1;Case<=Total;Case++){H=read(),W=read(),memset(a,0,sizeof(a)),memset(b,0,sizeof(b));for (int i=0;i<num;i++)a2[i].init(),b2[i].init();for (int i=0;i<H;i++)scanf("%s",T[i]);for (int i=0;i<H;i++)for (int j=0;j<W;j++)if (T[i][j]=='G') a2[R*C-1-i*C-j]=comp(1,0);else b2[R*C-1-i*C-j]=comp(1,0);FFT(a2,1),FFT(b2,1);for (int i=0;i<num;i++)a2[i]=a1[i]*a2[i],b2[i]=b1[i]*b2[i];FFT(a2,-1),FFT(b2,-1);for (int i=0;i<R-H;i++)for (int j=0;j<C-W;j++)a[i][j]=(int)(a2[i*C+j+R*C-1].r+0.5),b[i][j]=(int)(b2[i*C+j+R*C-1].r+0.5);int x,y;x=y=0;for (int i=0;i<R-H;i++)for (int j=0;j<=C-W;j++)if (a[i][j]+b[i][j]>a[x][y]+b[x][y]) x=i,y=j;printf("Case #%d: %d %d %d %d\n",Case,x+1,y+1,a[x][y],b[x][y]);}}