hdu 5147 树状数组

标签：

http://acm.hdu.edu.cn/showproblem.php?pid=5147

1
5
1 3 2 4 5

4

/**
hdu5147 树状数组
解题思路：
        要统计四元组的数量我们可以通过枚举c，然后统计区间[1,c-1]有多少二元组(a,b)满足a<b且Aa<Ab，以及统计出区间[c+1,n]有多少d满足Ac<Ad，
根据乘法原理，把这两项乘起来就可以统计到答案里了.然后我们来处理子问题：区间[1,c-1]内有多少二元组(a,b).那么我们可以枚举b，然后统计
区间[1,b-1]内有多少a满足Aa<Ab，那么这个可以通过用树状数组询问前缀和来实现.

具体实现：b[i]和c[i]中存储的分别为以i结尾的Ax<Ay的对数和从i+1到n中Ax<Ay的对数，二者相乘即为答案。
时间复杂度是O(nlogn).
*/
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
typedef long long LL;

int C[100005],b[100005],c[100005],a[100005];
int n;

int lowbit(int x)
{
    return x&(-x);
}
int sum(int x)
{
    int ret=0;
    while(x>0)
    {
        ret+=C[x];
        x-=lowbit(x);
    }
    return ret;
}
void add(int x,int d)
{
    while(x<=n)
    {
        C[x]+=d;
        x+=lowbit(x);
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        memset(b,0,sizeof(b));
        memset(c,0,sizeof(c));
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(C,0,sizeof(C));
        for(int i=1;i<=n;i++)
        {
            b[i]=sum(a[i]);
            add(a[i],1);
        }
        memset(C,0,sizeof(C));
        for(int i=n;i>=1;i--)
        {
            c[i]=sum(n)-sum(a[i])+c[i+1];
            add(a[i],1);
        }
        LL ans=0;
        for(int i=2;i<=n-2;i++)
        {
            LL t1=b[i];
            LL t2=c[i+1];
            ans+=t1*t2;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}

hdu 5147 树状数组

标签：

原文地址：http://blog.csdn.net/lvshubao1314/article/details/42061753