BZOJBZOJBZOJ看不了权限题怎么办?
把网址中的problemproblemproblem改成showshowshow就可以了(虽然没法提交)
可以从darkbzojdarkbzojdarkbzoj上下数据啊
考虑处理222个东西
l[i][j]l[i][j]l[i][j]表示iii前面,值为jjj的个数,r[i][j]r[i][j]r[i][j]表示iii后面,值为jjj的个数
则ans=∑i=1n∑j=0max(a)l[i][j]∗r[i][2∗a[i]−j]ans=\sum_{i=1}^{n}\sum_{j=0}^{max(a)}l[i][j]*r[i][2*a[i]-j]ans=∑i=1n∑j=0max(a)l[i][j]∗r[i][2∗a[i]−j]
发现这是一个卷积的形式,但直接fftfftfft也不行
考虑分块对于当前块统计这个块前的lll,块后的rrr
对于有在块内的再重新统计
块大小大约在nlogn\sqrt {nlogn}nlogn最佳
复杂度O(nnlogn)O(n\sqrt{nlogn})O(nnlogn)
#include<bits/stdc++.h>
using namespace std;
#define ll long long
inline int read(){
char ch=getchar();
int res=0,f=1;
while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=getchar();
return res*f;
}
const int N=100005;
const int M=120005;
const double pi=acos(-1);
struct plx{
double x,y;
plx(double _x=0,double _y=0):x(_x),y(_y){}
friend inline plx operator +(const plx &a,const plx &b){
return plx(a.x+b.x,a.y+b.y);
}
friend inline plx operator -(const plx &a,const plx &b){
return plx(a.x-b.x,a.y-b.y);
}
friend inline plx operator *(const plx &a,const plx &b){
return plx(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);
}
}A[N],B[N];
int rev[M],lim=1,tim=0;
inline void fft(plx f[],int kd){
for(int i=0;i<lim;i++)if(i<rev[i])swap(f[i],f[rev[i]]);
for(int mid=1;mid<lim;mid<<=1){
plx now=plx(cos(pi/mid),kd*sin(pi/mid));
for(int i=0;i<lim;i+=(mid<<1)){
plx w=plx(1,0);
for(int j=0;j<mid;j++,w=w*now){
plx a0=f[i+j],a1=w*f[i+j+mid];
f[i+j]=a0+a1,f[i+j+mid]=a0-a1;
}
}
}
if(kd==-1)for(int i=0;i<lim;i++)f[i].x/=lim;
}
int n,a[N],L[N],R[N],mx,cnt,blo,l[M],r[M];
ll ans;
int main(){
n=read(),blo=2000,cnt=(n-1)/blo+1;
for(int i=1;i<=n;i++)a[i]=read(),r[a[i]]++,mx=max(mx,a[i]);
while(lim<=mx*2)lim<<=1,tim++;
for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(tim-1));
for(int i=1;i<=cnt;i++)L[i]=(i-1)*blo+1,R[i]=i*blo;R[cnt]=n;
for(int i=1;i<=cnt;i++){
for(int j=L[i];j<=R[i];j++)r[a[j]]--;
for(int j=0;j<lim;j++)A[j]=B[j]=plx(0,0);
for(int j=1;j<=mx;j++)A[j]=plx(l[j],0),B[j]=plx(r[j],0);
fft(A,1),fft(B,1);
for(int j=0;j<lim;j++)A[j]=A[j]*B[j];
fft(A,-1);
for(int j=L[i];j<=R[i];j++)ans+=(ll)(A[2*a[j]].x+0.5);
for(int j=L[i];j<=R[i];j++){
for(int k=L[i];k<j;k++)
if(2*a[j]-a[k]>0)ans+=r[2*a[j]-a[k]];
for(int k=j+1;k<=R[i];k++)
if(2*a[j]-a[k]>0)ans+=l[2*a[j]-a[k]];
l[a[j]]++;
}
}
cout<<ans<<'\n';
}