hdu 5147 树状数组

题意：求满足a<b<c<d，A[a]<A[b]，A[c]<A[d]的所有四元组(a,b,c,d)的个数

 

看到逆序对顺序对之类的问题一开始想到了曾经用归并排序求逆序对，结果YY了半天无果而终。

其实用树状数组做也很方便。

比如对于序列1 3 2 4 5，设数组c：0 0 0 0 0

 

一开始读入1，add(a[1],1)，c：1 0 0 0 0

读入3，add(a[2],1)，c：1 0 1 0 0

这时sum(a[2])-1就是3前面比3小的数的个数。

下面同理，

读入2，add(a[3],1)，c：1 1 1 0 0

这时sum(a[3])-1就是2前面比2小的数的个数。

以此类推。

 

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
int c[50010],f[50010],a[50010],g[50010];
int n,T;

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

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

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

int main()
{
    scanf("%d\n",&T);
    while (T--)
    {
        scanf("%d\n",&n);
        for (int i=1; i<=n; i++)
            scanf("%d\n",&a[i]);

        memset(c,0,sizeof(c));
        for(int i=1; i<=n; i++)
        {
            add(a[i],1);
            f[i]=sum(a[i])-1;
        }

        memset(c,0,sizeof(c));
        for(int i=n; i>=1; i--)
        {
            add(a[i],1);
            g[i]=n-i+1-sum(a[i]);
        }

        long long ans = 0;
        long long sum = 0;
        for(int i = 1; i <= n; i++)
        {
            ans += sum*g[i];
            sum += f[i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}