C++ | Leetcode C++题解之第493题翻转对
题目:
题解:
class BIT {
private:vector<int> tree;int n;public:BIT(int _n) : n(_n), tree(_n + 1) {}static constexpr int lowbit(int x) {return x & (-x);}void update(int x, int d) {while (x <= n) {tree[x] += d;x += lowbit(x);}}int query(int x) const {int ans = 0;while (x) {ans += tree[x];x -= lowbit(x);}return ans;}
};class Solution {
public:int reversePairs(vector<int>& nums) {set<long long> allNumbers;for (int x : nums) {allNumbers.insert(x);allNumbers.insert((long long)x * 2);}// 利用哈希表进行离散化unordered_map<long long, int> values;int idx = 0;for (long long x : allNumbers) {values[x] = ++idx;}int ret = 0;BIT bit(values.size());for (int i = 0; i < nums.size(); i++) {int left = values[(long long)nums[i] * 2], right = values.size();ret += bit.query(right) - bit.query(left);bit.update(values[nums[i]], 1);}return ret;}
};