hdu2689Sort it树状数组

来源:互联网 发布:凯立德端口文件 编辑:程序博客网 时间:2024/06/04 18:01
思路:树状数组:
例子:输入: 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 1100int 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;}


原创粉丝点击