找规律 calc

来源:互联网 发布:怎么报送发票数据 编辑:程序博客网 时间:2024/06/03 06:59

 

      看一个数对答案的贡献,就是找他前面有多少比它小的,后面有多少比他大的,一乘,就是他前后有多少涉及到他的顺序对,就是他对答案的贡献。

     因为n<=300000,效率N*log(N)能过,也就是树状数组。离散一下权值,用权值作为数组下标。

      维护两个,一个是之前最小,正推,每推一个,压进当前位。另一个反推,找到比自己小的,做减法即可。

    

#include<cstdio>#include<cstdlib>#include<cstring>#include<iostream>#include<algorithm>#define ll long longusing namespace std;int read(){int sum=0,f=1;char x=getchar();while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}while(x>='0'&&x<='9'){sum=sum*10+x-'0';x=getchar();}return sum*f;}int n,a[300205],h[300205],sz;ll ans=0;int t1[300205],t2[300205],v1[300205],v2[300205];int lowbit(int x){return x&(-x);}void add1(int x,int k){for(int i=x;i<=n;i+=lowbit(i))t1[i]+=k;}void add2(int x,int k){for(int i=x;i<=n;i+=lowbit(i))t2[i]+=k;}ll q1(int x){int s=0;while(x>0){s+=t1[x],x-=lowbit(x);}return s;}ll q2(int x){int s=0;while(x>0){s+=t2[x],x-=lowbit(x);}return s;}int main(){//freopen("calc.in","r",stdin);//freopen("calc.out","w",stdout);n=read();for(int i=1;i<=n;i++)h[i]=a[i]=read();sort(h+1,h+n+1);for(int i=1;i<=n;i++)a[i]=lower_bound(h+1,h+n+1,a[i])-h;for(int i=1;i<=n;i++){v1[i]=q1(a[i]-1);add1(a[i],1);}for(int i=n;i>=1;i--){int k=n-i;v2[i]=k-q2(a[i]);add2(a[i],1);}for(int i=1;i<=n;i++)ans+=v1[i]*v2[i];printf("%lld\n",ans);}