树状数组——楼兰图腾

传送门：241. 楼兰图腾 - AcWing题库

思路：

以v为例，设left[i]是元素a[i]左边的元素中比它大的元素的个数，设right[i]是元素a[i]右边的元素中比它小的元素的个数。v的个数很明显等于所有left[i]*right[i]的和。

^的情况类似，不再解释。

而left数组和right数组的构造可以采用树状数组求逆序对的方法来求的。

树状数组——逆序对（范围较小）_北岭山脚鼠鼠的博客-CSDN博客

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<stack>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int m;
int c[N];
int a[N];
int l[N];
int r[N];
int n;
void add(int x,int y) //初始化操作，对于每个a[i]做一遍add(i,a[i]);
{
    for(;x<=N;x+=x&(-x)) c[x]+=y;
}
int lowbit(int x)
{
    return (x&(-x));
}
int ask(int x) //查询操作，[1,x]的前缀和
{
    int ans=0;
    for(;x;x-=x&(-x)) ans+=c[x];
    return ans;
}
void get_l(int flag)
{
    memset(c,0,sizeof c);
    if(flag==1)
    {
        for(int i=1;i<=n;i++)
        {
            l[i]=ask(m)-ask(a[i]-1);
            add(a[i],1);
        }
    }else
    {
         for(int i=1;i<=n;i++)
        {
            l[i]=ask(a[i]-1);
            add(a[i],1);
        }
    }
}
void get_r(int flag)
{
    memset(c,0,sizeof c);
    if(flag==1)
    {
        for(int i=n;i;i--)
        {
            r[i]=ask(m)-ask(a[i]-1);
            add(a[i],1);
        }
    }else
    {
         for(int i=n;i;i--)
        {
            r[i]=ask(a[i]-1);
            add(a[i],1);
        }
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    m=max(m,a[i]);
    }
    ll ans1=0,ans2=0;
    get_l(1);
    get_r(1);
    for(int i=1;i<=n;i++)
        ans1+=(ll)l[i]*r[i];
    get_l(0);
    get_r(0);
    for(int i=1;i<=n;i++)
        ans2+=(ll)l[i]*r[i];

    cout<<ans1<<' '<<ans2<<endl;
    return 0;
}