AcWing 241. 楼兰图腾

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
typedef long long ll;
ll ans,l[N],r[N],a[N],c[N],n,m,mx;
ll lowbit(ll x){return x&(-x);}
void add(ll x,ll ad){while(x<=n)c[x]+=ad,x+=lowbit(x);}
ll ask(ll x){ans=0;while(x)ans+=c[x],x-=lowbit(x);return ans;}
void getv()
{
    for(ll i=1;i<=n;i++) l[i]=ask(mx)-ask(a[i]),add(a[i],1);
    memset(c,0,sizeof c);
    for(ll i=n;i;i--) r[i]=ask(mx)-ask(a[i]),add(a[i],1);
}
void getn()
{
    memset(c,0,sizeof c);
    for(ll i=1;i<=n;i++)l[i]=ask(a[i]),add(a[i],1);
    memset(c,0,sizeof c);
    for(ll i=n;i;i--)r[i]=ask(a[i]),add(a[i],1);
}
ll sum()
{
    ans=0;
    for(ll i=1;i<=n;i++)ans+=l[i]*r[i];
    return ans;
}
int main()
{
    cin>>n;
    for(ll i=1;i<=n;i++)cin>>a[i],mx=max(mx,a[i]);
    getv();cout<<sum()<<' ';
    getn();cout<<sum()<<endl;
    return 0;
}