树状数组专题(三)-Count Inversions

最新推荐文章于 2021-08-04 19:48:17 发布

WUST_XIAO

最新推荐文章于 2021-08-04 19:48:17 发布

阅读量291

点赞数

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/xr469786706/article/details/88068471

版权

数据结构与算法专栏收录该内容

67 篇文章 1 订阅

订阅专栏

树状数组专题(三)-Count Inversions

在这里插入图片描述

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;

const int maxn = 10000;
int c[maxn];
int a[maxn];
int lowbit(int x){
    return x & (-x);
}

void update(int x,int v){
    for(int i = x ; i < maxn ; i += lowbit(i)){
        c[i] += v;
    }
}

int getSum(int x){
    int sum = 0;
    for(int i = x ; i > 0 ; i -= lowbit(i)){
        sum += c[i];
    }
    return sum;
}

int main(){
    freopen("in.txt","r",stdin);
    char str[1000];
    while(cin.getline(str,1000)){
        memset(c,0,sizeof(c));
        int k = 1;
        char *p = strtok(str," ");
        while(p != NULL){
            a[k++] = atoi(p);
            p = strtok(NULL," ");
        }
        int ans = 0;
        for(int i = 1 ; i < k ; i++){
            update(a[i],1);
            ans += i - getSum(a[i]);
        }
        printf("%d\n",ans);
    }
    return 0;
}