离散化+树状数组

题目描述

Erwin最近对一种叫"thair"的东西巨感兴趣。。。

在含有n个整数的序列a1,a2…an中，

三个数被称作"thair"当且仅当i<j<k且ai<aj<ak

求一个序列中"thair"的个数。
输入格式

开始一个正整数n,

以后n个数a1~an。
输出格式

"thair"的个数

思路：枚举中间的数字，求排在它前面，小于它的有几个。
排在它后面，大于它的有几个。使用两次树状数组就可以了。
坑点：数组开long long，这样每次用for清空数组。memset没办法使用。

#include<bits/stdc++.h>
#define MAXN 31000
using namespace std;
typedef long long LL;

LL c[MAXN];
LL a[MAXN],b[MAXN],h[MAXN];
int n,m;

int lowbit(int x)
{
    return x&(-x);
}

void add(int x,LL d)
{
    while(x<=n)
    {
        c[x]+=d;
        x+=lowbit(x);
    }
}

LL sum(int x)
{
    LL ans=0;
    while(x>0)
    {
        ans+=c[x];
        x-=lowbit(x);
    }
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);b[i]=a[i];
    }
    sort(b+1,b+1+n);
    int mm=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++)
    {
        int pos=lower_bound(b+1,b+1+mm,a[i])-b;
        h[i]=sum(pos-1);add(pos,1);
    }
    LL ans=0;
    for(int i=1;i<=n;i++)
        c[i]=0;
    for(int i=n;i>=1;i--)
    {
        int pos=lower_bound(b+1,b+1+mm,a[i])-b;
        ans+=h[i]*(sum(mm)-sum(pos));add(pos,1);
    }
    printf("%lld\n",ans);
    return 0;
}