BZOJ3509: [CodeChef] COUNTARI

3509: [CodeChef] COUNTARI

Time Limit: 40 Sec  Memory Limit: 128 MB
Submit: 339  Solved: 85
[Submit][Status]

Description

给定一个长度为N的数组A[]，求有多少对i, j, k（1<=i<j<k<=N）满足A[k]-A[j]=A[j]-A[i]。

Input

第一行一个整数N（N<=10^5）。
接下来一行N个数A[i]（A[i]<=30000）。

Output

一行一个整数。

Sample Input

10
3 5 3 6 3 4 10 4 5 2

Sample Output

9

题解：

Mektpoy：

分成N/size块

对于三个数在同一块内

可以暴力size^2

对于两个数在同一块内

可以和上面一起暴力size^2

对于中间的数在当前块内 别的两个数不在当前块内

将两边卷积value*log(value)

时间复杂度 O(N*size+N/size*value*log(value))

size取2600左右可以达到比较满意的时间

感觉n*size已经超了。。。居然可过。。。orz
代码参考了云神的，感觉自己又没代码能力，思路又捉鸡，还怎么混啊。。。
代码：

 #include<cstdio>

 #include<cstdlib>

 #include<cmath>

 #include<cstring>

 #include<algorithm>

 #include<iostream>

 #include<vector>

 #include<map>

 #include<set>

 #include<queue>

 #include<string>

 #define inf 1000000000

 #define maxn 500000+5

 #define maxm 20000000+5

 #define eps 1e-10

 #define ll long long

 #define pa pair<int,int>

 #define for0(i,n) for(int i=0;i<=(n);i++)

 #define for1(i,n) for(int i=1;i<=(n);i++)

 #define for2(i,x,y) for(int i=(x);i<=(y);i++)

 #define for3(i,x,y) for(int i=(x);i>=(y);i--)

 #define mod 1000000007

 using namespace std;

 inline int read()

 {

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

     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}

     while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}

     return x*f;

 }
 struct cp
 {
     double x,y;
     cp operator +(cp b){return (cp){x+b.x,y+b.y};}
     cp operator -(cp b){return (cp){x-b.x,y-b.y};}
     cp operator *(cp b){return (cp){x*b.x-y*b.y,x*b.y+y*b.x};}
 };
 int n,m,len,mx,num[maxn],l[maxn],r[maxn],st[maxn],ed[maxn],size,rev[maxn];
 cp a[maxn],b[maxn],c[maxn],y[maxn];
 const double PI=acos(-1.0);
 void fft(cp *x,int n,int flag)
 {
     for0(i,n-)y[rev[i]]=x[i];
     for0(i,n-)x[i]=y[i];
     for(int m=;m<=n;m<<=)
     {
         cp wn=(cp){cos(2.0*PI/m*flag),sin(2.0*PI/m*flag)};
         for(int i=;i<n;i+=m)
         {
             cp w=(cp){,};int mid=m>>;
             for0(j,mid-)
             {
                 cp u=x[i+j],v=x[i+j+mid]*w;
                 x[i+j]=u+v;x[i+j+mid]=u-v;
                 w=w*wn;
             }
         }
     }
     if(flag==-)for0(i,n-)x[i].x/=n;
 }            

 int main()

 {

     freopen("input.txt","r",stdin);

     freopen("output.txt","w",stdout);

     n=read();
     for1(i,n)num[i]=read(),mx=max(num[i],mx),r[num[i]]++;
     mx++;mx=mx*-;m=;
     while(m<=mx)m<<=,len++;mx=m;
     for0(i,mx-)
     {
         int x=i,y=;
         for1(j,len)y<<=,y|=x&,x>>=;
         rev[i]=y;
     }
     size=;m=(n-)/size+;
     for1(i,m)st[i]=ed[i-]+,ed[i]=i*size;
     ed[m]=n;
     ll ans=;
     for1(i,m)
     {
      for2(j,st[i],ed[i])r[num[j]]--;
      for0(j,mx-)a[j]=(cp){l[j],};
      for0(j,mx-)b[j]=(cp){r[j],};
      fft(a,mx,);fft(b,mx,);
      for0(j,mx-)c[j]=a[j]*b[j];
      fft(c,mx,-);
      for2(j,st[i],ed[i])ans+=(ll)(c[*num[j]].x+0.5);
      for2(j,st[i],ed[i])
       {
         for2(k,st[i],j-)if(*num[j]-num[k]>=)ans+=r[*num[j]-num[k]];
         for2(k,j+,ed[i])if(*num[j]-num[k]>=)ans+=l[*num[j]-num[k]];
         l[num[j]]++;
       }
     }
     cout<<ans<<endl;

     return ;

 }  

一堆j打成i我也是醉了。。。