hdu5147 Sequence II树状数组求逆序对

来源:互联网 发布:mac pages怎么保存 编辑:程序博客网 时间:2024/04/29 13:33
//用树状数组求出在b前面比b小的数的个数
//然后求b后面的顺序对的个数,
//枚举b可得quad
//由于数列是从1到n的所有数
//那么(n-num[j])-(j-1-totol[j])即为第j个数之后比j大的数的个数
//其中num[j]表示第j个数,total[j]表示在j之前比j小的数的个数
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=50010;
int tree[maxn];
int num[maxn];
__int64 total[maxn];
int lowbit(int i)
{
    return i&(-i);
}
int getsum(int i)
{
    int sum=0;
    while(i>0)
    {
        sum+=tree[i];
        i-=lowbit(i);
    }
    return sum;
}
void update(int i,int dx)
{
    while(i<maxn)
    {
        tree[i]+=dx;
        i+=lowbit(i);
    }
}
int main()
{
    int T;int n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int i;int j;
        memset(tree,0,sizeof(tree));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
            total[i]=getsum(num[i]);
            update(num[i],1);
        }
        __int64 ans=0;
        __int64 tear[maxn];//tear[j]表示从j开始到最后的逆序对的个数
        memset(tear,0,sizeof(tear));
        for(j=n-1;j>=1;j--)
        tear[j]=tear[j+1]+(n-j+1-num[j]+total[j]);
        for(i=1;i<=n;i++)
                ans+=(total[i])*tear[i+1];
       printf("%I64d\n",ans);
    }
    return 0;
}

































0 0
原创粉丝点击