BZOJ3509: [CodeChef] COUNTARI
2014-12-17 18:33
477 查看
3509: [CodeChef] COUNTARI
Time Limit: 40 Sec Memory Limit: 128 MBSubmit: 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
103 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=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=10*x+ch-'0';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-1)y[rev[i]]=x[i]; for0(i,n-1)x[i]=y[i]; for(int m=2;m<=n;m<<=1) { cp wn=(cp){cos(2.0*PI/m*flag),sin(2.0*PI/m*flag)}; for(int i=0;i<n;i+=m) { cp w=(cp){1,0};int mid=m>>1; for0(j,mid-1) { 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==-1)for0(i,n-1)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*2-1;m=1; while(m<=mx)m<<=1,len++;mx=m; for0(i,mx-1) { int x=i,y=0; for1(j,len)y<<=1,y|=x&1,x>>=1; rev[i]=y; } size=2000;m=(n-1)/size+1; for1(i,m)st[i]=ed[i-1]+1,ed[i]=i*size; ed[m]=n; ll ans=0; for1(i,m) { for2(j,st[i],ed[i])r[num[j]]--; for0(j,mx-1)a[j]=(cp){l[j],0}; for0(j,mx-1)b[j]=(cp){r[j],0}; fft(a,mx,1);fft(b,mx,1); for0(j,mx-1)c[j]=a[j]*b[j]; fft(c,mx,-1); for2(j,st[i],ed[i])ans+=(ll)(c[2*num[j]].x+0.5); for2(j,st[i],ed[i]) { for2(k,st[i],j-1)if(2*num[j]-num[k]>=0)ans+=r[2*num[j]-num[k]]; for2(k,j+1,ed[i])if(2*num[j]-num[k]>=0)ans+=l[2*num[j]-num[k]]; l[num[j]]++; } } cout<<ans<<endl; return 0; }
View Code
一堆j打成i我也是醉了。。。
相关文章推荐
- bzoj 3509: [CodeChef] COUNTARI] [分块 生成函数]
- BZOJ3509: [CodeChef] COUNTARI
- BZOJ3509 [CodeChef] COUNTARI
- bzoj 3509: [CodeChef] COUNTARI 分块fft
- bzoj 3509 [CodeChef] COUNTARI
- bzoj3509[CodeChef]COUNTARI
- [bzoj3509][codechef]COUNTARI
- [bzoj3509][CodeChef]COUNTARI
- BZOJ 3509 [CodeChef] COUNTARI ——分块 FFT
- BZOJ 3509: [CodeChef] COUNTARI
- BZOJ[3509][CodeChef] COUNTARI 分块+FFT
- 【 bzoj 3509 】[CodeChef] COUNTARI - 分块FFT
- bzoj 3509: [CodeChef] COUNTARI (FFT+分块)
- [生成函数 FFT 分块] BZOJ 3509 [CodeChef] COUNTARI
- BZOJ 3509: [CodeChef] COUNTARI
- BZOJ3509: [CodeChef] COUNTARI
- [BZOJ3509][CodeChef][FFT][分块]COUNTARI
- [CODECHEF]COUNTARI
- BZOJ 3509 分块FFT
- Bzoj3509:[CodeChef] COUNTARI:分块+FFT