思路:树状数组:
例子:输入: 4, 3, 2 ,1
下标: 1,2, 3, 4
由于求逆序对,所以我们看到给出数列4,3, 2, 1下标正好是他们的正序对
所以使用树状数组算法以下标为中心:
当输入4之前,此时tree[]数组为{0, 0, 0, 0},我们获得此时逆序个数getSum(n) - getSum(4)= 0 - 0 =0,把数组的第4位加1
当输入3之前,此时tree[]数组为{0, 0, 0, 1},我们获得此时逆序个数getSum(n) - getSum(3)= 1 - 0 = 1,把数组的第3位加1
当输入2之前,此时tree[]数组为{0, 0, 1, 2},我们获得此时逆序个数getSum(n) - getSum(2)= 2 - 0 =2,把数组的第2位加1
当输入3之前,此时tree[]数组为{0, 1, 2, 3},我们获得此时逆序个数getSum(n) - getSum(1)= 3 - 0 =3,把数组的第1位加1
此时tree[]数组为{1, 2, 3, 4},所以总共逆序对为0 + 1 + 2 + 3 = 6;
例子:输入: 4, 3, 2 ,1
下标: 1,2, 3, 4
由于求逆序对,所以我们看到给出数列4,3, 2, 1下标正好是他们的正序对
所以使用树状数组算法以下标为中心:
当输入4之前,此时tree[]数组为{0, 0, 0, 0},我们获得此时逆序个数getSum(n) - getSum(4)= 0 - 0 =0,把数组的第4位加1
当输入3之前,此时tree[]数组为{0, 0, 0, 1},我们获得此时逆序个数getSum(n) - getSum(3)= 1 - 0 = 1,把数组的第3位加1
当输入2之前,此时tree[]数组为{0, 0, 1, 2},我们获得此时逆序个数getSum(n) - getSum(2)= 2 - 0 =2,把数组的第2位加1
当输入3之前,此时tree[]数组为{0, 1, 2, 3},我们获得此时逆序个数getSum(n) - getSum(1)= 3 - 0 =3,把数组的第1位加1
此时tree[]数组为{1, 2, 3, 4},所以总共逆序对为0 + 1 + 2 + 3 = 6;
#include <iostream>
using namespace std;
#define NSIZ 1100
int tree[NSIZ];
int n;
int lowbit(int x)
{
return x&(-x);
}
void update(int i, int x)
{
while(i <= n)
{
tree[i] += x;
i += lowbit(i);
}
}
int getSum(int i)
{
int res = 0;
while(i >= 1)
{
res += tree[i];
i -= lowbit(i);
}
return res;
}
int main()
{
int i , t, sum = 0;
while(scanf("%d", &n) != EOF)
{
sum = 0;
memset(tree, 0, sizeof(tree));
for (i = 1;i <= n; ++i)
{
scanf("%d", &t);
sum += getSum(n) - getSum(t);
update(t, 1);
}
printf("%d\n", sum);
}
return 0;
}