BZOJ2837 : 小强的形状
离散化后通过树状数组求出:
b[i]为i之前比它小的。
c[i]为i之前比它大的=i-1-i之前小于等于它的。
d[i]为i之后比它小的。
e[i]为i之后比它大的=n-i-f[i]。
f[i]为i之后小于等于它的。
则:
$cnt_{123}=\sum_{i=1}^n b[i]e[i]$
$cnt_{321}=\sum_{i=1}^n c[i]d[i]$
$cnt_{213}$=将序列a翻转后的$cnt_{312}$
$cnt_{231}$=将序列a翻转后的$cnt_{132}$
$cnt_{312}=\sum_{i=1}^n b[i]c[i]-cnt_{132}$
$cnt_{132}$可以这样计算:
枚举$3$的位置$i$,它右边比它小的$k$有$b[k]$的贡献,还要减去$i$右下角内$12$形状的二元组的个数,即右下角$b[k]+f[k]$的和$-C_{d[i]}^2$。
时间复杂度$O(n\log n)$。
#include<cstdio> #include<algorithm> #define N 100010 typedef long long ll; int n,i,a[N],b[N],c[N],d[N],e[N],f[N];ll bit[N],c1,c2,c3,c4,c5,c6,all; inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';} inline int lower(int x){ int l=1,r=n,mid,t; while(l<=r)if(b[mid=(l+r)>>1]<=x)l=(t=mid)+1;else r=mid-1; return t; } inline void add(int x,int y){for(;x<=n;x+=x&-x)bit[x]+=y;} inline ll ask(int x){ll t=0;for(;x;x-=x&-x)t+=bit[x];return t;} void write(ll x){for(printf("0."),i=0;i<20;i++)x*=10,printf("%d",x/all),x%=all;puts("");} int main(){ for(read(n),i=1;i<=n;i++)read(a[i]),b[i]=a[i]; for(std::sort(b+1,b+n+1),i=1;i<=n;i++)a[i]=lower(a[i]); for(i=1;i<=n;i++)b[i]=ask(a[i]-1),c[i]=i-1-ask(a[i]),add(a[i],1); for(i=1;i<=n;i++)bit[i]=0; for(i=n;i;i--)d[i]=ask(a[i]-1),f[i]=ask(a[i]),e[i]=n-i-f[i],add(a[i],1); for(i=1;i<=n;i++)bit[i]=0; for(i=n;i;i--){ c1+=1LL*b[i]*e[i],c6+=1LL*c[i]*d[i]; c5+=1LL*b[i]*c[i],c2+=ask(a[i]-1)-1LL*d[i]*(d[i]-1)/2,add(a[i],b[i]+f[i]); } c5-=c2; for(i=1;i<n-i+1;i++)std::swap(a[i],a[n-i+1]); for(i=1;i<=n;i++)bit[i]=0; for(i=1;i<=n;i++)b[i]=ask(a[i]-1),c[i]=i-1-ask(a[i]),add(a[i],1); for(i=1;i<=n;i++)bit[i]=0; for(i=n;i;i--)d[i]=ask(a[i]-1),f[i]=ask(a[i]),add(a[i],1); for(i=1;i<=n;i++)bit[i]=0; for(i=n;i;i--)c3+=1LL*b[i]*c[i],c4+=ask(a[i]-1)-1LL*d[i]*(d[i]-1)/2,add(a[i],b[i]+f[i]); c3-=c4; all=c1+c2+c3+c4+c5+c6; write(c1),write(c2),write(c3),write(c4),write(c5),write(c6); return 0; }