树状数组求逆序对的方法(学习总结)
逆序对:i<j,a[i]>a[j]
树状数组求逆序对:先把tree置0,每次输入一个数x,用1去更新tree
例如:
tree:0 0 0 0 0
x=4:
tree: 0 0 0 0 1
然后对x进行查询read(x),可以得到x之前插入了多少个数,因为x之前插入的数都比x小,所以用当前插入的数的总数i减去read(x)可以得到当前插入的i个数中有多少个比x大,即x有多个逆序对。
离散化: 如果插入的x很大,每次对x进行查询,时间用的多,且tree的空间不够。这时候需要进行离散化。具体做法是每次输入一个数的同时记录下该数输入的顺序是多少。然后对输入的数进行排序。
例如:
输入:9 1 0 5 4
id: 1 2 3 4 5
排序:0 1 4 5 9
id1: 3 2 5 4 1
3 2 5 4 1 和9 1 0 5 4 的逆序对都是6
#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) (x)&(-x)
const int maxn=1e5;
using namespace std;
ll n,tree[maxn];
struct node
{
int val;
int id;
}arr[maxn];
bool cmp(node a,node b)
{
return a.val<b.val;
}
void update(int k,int a)
{
while(k<=n)
{
tree[k]+=a;
k+=lowbit(k);
}
}
ll read(ll k)
{
ll sum=0;
while(k)
{
sum+=tree[k];
k-=lowbit(k);
}
return sum;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>arr[i].val;
arr[i].id=i;
}
sort(arr+1,arr+n+1,cmp);
ll ans=0;
for(int i=1;i<=n;i++)
{
update(arr[i].id,1);
ans+=(i-read(arr[i].id));
//cout<<arr[i].id<<" ";
}
cout<<ans<<endl;
return 0;
}
版权声明:本文为qq_43022818原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。