每次查询是求从查询节点开始(包括查询节点)自右向左层数尽可能低的点的权值的和,算过的点的根节点不再计算(也就是从查询节点开始,从右到左所有需要计算的点的权值和),得到的是前缀和。
每次修改是修改以该点为子节点的所有节点的值。
查询和修改都是以跳跃的形式进行的,时间复杂度都为logn。
树状数组可以以nlogn的时间复杂度求序列的逆序对
(本质:通过数组自身的有序加上循环操作时的有序,可以同时实现两个维度上的有序)
POJ 2299 https://vjudge.net/contest/301590#problem/A
设A为有n个数字的有序集(n>1),其中所有数字各不相同。如果存在正整数i, j使得1 ≤iA[j],则这个有序对称为A的一个逆序对,也称作逆序数。
在这个问题中,你需要快速的求出一个给定数组中逆序对的数量
Input
输入包含多个测试用例,每个测试用例第一行为一个整数n(n<500000)表示数组的长度,下面n行每一行包含一个整数0≤a[i]≤999,999,999,即数组中第i个元素的值。当n=0时结束输入
Output
对于输入的每个数组,程序应该输出该数组中逆序对的数量
Sample Input
5
9
1
0
5
4
3
1
2
3
0
Sample Output
6
0
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int num=500005;
struct cod
{
int num,ind;
}codzu[num];
int zuc[num],n;
int lowbit(int a)
{
return a&-a;
}
void change(int a,int val)
{
while(a<=n){
zuc[a]+=val;
a+=lowbit(a);
}
}
int ask(int a)
{
int sum=0;
while(a>0){
sum+=zuc[a];
a-=lowbit(a);
}
return sum;
}
bool cmp(cod a,cod b)
{
return a.num<b.num;
}
int main()
{
while(scanf("%d",&n)!=-1){
if(n==0)
break;
for(int i=1;i<=n;i++)
zuc[i]=lowbit(i);
for(int i=1;i<=n;i++){
int a;
scanf("%d",&a);
codzu[i].num=a;
codzu[i].ind=i;
}
sort(codzu+1,codzu+n+1,cmp);
long long ans=0;
for(int i=1;i<=n;i++){
change(codzu[i].ind,-1);
ans+=ask(codzu[i].ind);
}
printf("%lld\n",ans);
}
return 0;
}