树状数组求逆序数(模板)

转载请注明出处:http://blog.csdn.net/u012860063?viewmode=contents

#include<cstdio>
#include<iostream>
#include<cstring>
using  namespace std;
const int MAX = 100005;
int a[MAX] , cnt[MAX] , leftMax[MAX] ;
int leftMin[MAX] ,rightMax[MAX] , rightMin[MAX];

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

void update(int x , int y)
{
    while(x <= MAX)
    {
        cnt[x] += y;
        x += lowbit(x);
    }
}

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

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

        memset(cnt , 0 , sizeof(cnt));//清零
        for( i = 1 ; i <= n ; i++)
        {
            update(a[i] , 1);
			leftMin[i] = sum(a[i]-1);//计算左边小的个数
            leftMax[i] = i - leftMin[i] -1;//计算左边大的个数
		//	leftMax[i] = sum(MAX) - sum(skill[i]);//计算左边大的个数
        }
		memset(cnt , 0 , sizeof(cnt));//注意此处要再次清零
        for( i = n,j = 1 ; i >= 1 ;j++, i--)
        {
            update(a[i] , 1);
			rightMin[i] = sum(a[i]-1);//计算右边小的个数
            rightMax[i] = j - rightMin[i]-1;//计算右边大的个数
		//	rightMax[i] = sum(MAX) - sum(skill[i]);//计算左右大的个数
        }
        __int64 sum = 0;
        for( i = 1 ; i <= n ; i++)
        {
            sum += leftMax[i]*rightMin[i] + leftMin[i]*rightMax[i];//交叉相乘取和
        }
        printf("%I64d\n" , sum);
    }
    return 0;
}


评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值