UVA1428树状数组

最新推荐文章于 2021-04-07 11:37:29 发布

翻译最新推荐文章于 2021-04-07 11:37:29 发布 · 475 阅读

UVA 专栏收录该内容

131 篇文章

订阅专栏

本文介绍了一种使用树状数组求解逆序对数量的方法。通过两次遍历数组，第一次从前往后插入并统计小于当前元素的数量，第二次从后往前插入并统计大于当前元素的数量，最终计算出所有逆序对的数量。

当时做的时候就是想的一边插入一边统计前后比它大的数的个数和比他小的个数，

其实就差了一个点没想到，只要再深入一点就做出来了，唉，看来一定要思考，深入思考。

先正向插入树状数组，一边插入一边统计，然后再反向插入一边，一边插入一边统计：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=100005;
int bit[N];
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int val)
{
while(x<N)
{
bit[x]+=val;
x+=lowbit(x);
}
}
int query(int x)
{
int ans=0;
while(x>0)
{
ans+=bit[x];
x-=lowbit(x);
}
return ans;
}
int main()
{
int t;
cin>>t;
while(t--)
{
int n;
cin>>n;
int a[N],rx[N],lx[N];
memset(bit,0,sizeof(bit));
for(int i=0;i<n;i++)
{
cin>>a[i];
lx[i]=query(a[i]);
add(a[i],1);
}
memset(bit,0,sizeof(bit));
for(int i=n-1;i>=0;i--)
{
rx[i]=query(a[i]);
add(a[i],1);
}
long long ans=0;
for(int i=0;i<n;i++)
ans+=lx[i]*(n-i-1-rx[i])+rx[i]*(i-lx[i]);
cout<<ans<<endl;
}
return 0;
}