剑指Offer——JZ35.数组中的逆序对【树状数组】

最新推荐文章于 2023-11-09 23:37:23 发布

原创最新推荐文章于 2023-11-09 23:37:23 发布 · 352 阅读

0 ·

CC 4.0 BY-SA版权

Linux 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种使用树状数组解决逆序对计算的高效算法。通过离散化处理，确保了编号逆序对与原始数据的一致性。代码示例展示了如何在C++中实现这一算法，包括添加元素、获取前缀和以及计算逆序对数量。

题目传送门

在这里插入图片描述

题解

非常简单的树状数组应用
需要离散化，离散后编号的的逆序对和之前相同

AC-Code

class Solution {
#define lowbit(x) (x&-x)
private:
    const int mod = 1e9 + 7;
    const static int maxn = 2e5 + 7;
    struct NODE {
        int val, pos;
        NODE(int v = 0, int p = 0):val(v), pos(p) {}
    }a[maxn];
    int c[maxn] = {0};
    
public:
    int InversePairs(vector<int> data) {
        int ans = 0;
        for(int i = 0; i < data.size(); ++i) {
            a[i + 1].pos = i + 1;
            a[i + 1].val = data[i];
        }
        sort(a + 1, a + 1 + data.size(), [](NODE a, NODE b)-> bool { return a.val < b.val; });
        for(int i = 1; i <= data.size(); ++i) {
            add(a[i].pos, 1);
            ans = (ans + i - getsum(a[i].pos)) % mod;
        }
        return ans;
    }
    void add(int i, int x) {
        while(i < maxn) {
            c[i] += x;
            i += lowbit(i);
        }
    }
    int getsum(int i) {
        int res = 0;
        while(i > 0) {
            res += c[i];
            i -= lowbit(i);
        }
        return res;
    }
 #undef lowbit(x)
};