Reverse Pairs

最新推荐文章于 2019-06-12 07:52:24 发布

转载最新推荐文章于 2019-06-12 07:52:24 发布 · 235 阅读

本文介绍了一种算法，用于计算数组中重要的逆序对数量。所谓重要的逆序对是指对于数组nums，若存在i<j且nums[i]>2*nums[j]的情况，则称(i,j)为一个重要逆序对。通过使用已排序数组来辅助计算，该算法能够有效地求解问题。

Given an array nums, we call (i, j) an important reverse pair if i < j and nums[i] > 2*nums[j].

You need to return the number of important reverse pairs in the given array.

Example1:

Input: [1,3,2,3,1]
Output: 2

Example2:

Input: [2,4,3,5,1]
Output: 3

class Solution {
public:
    int reversePairs(vector<int>& nums) {
        vector<int> sortedNums;
        int res = 0, target = 0, pos = 0;
        for (int i = nums.size() - 1; i >= 0; --i) {
            target = nums[i] > 0 ? (nums[i] - 1) / 2 : nums[i] / 2 - 1; 
            // find the target that is approximately nums[i]/2 in the sorted array
            auto ptr = upper_bound(sortedNums.begin(), sortedNums.end(), target);
            res += ptr - sortedNums.begin(); 
            // the number of reverse pair for nums[i] (j > i elements in nums are already sorted in "sortedNums")
            pos = upper_bound(sortedNums.begin(), sortedNums.end(), nums[i]) - sortedNums.begin(); 
            // insert nums[i] in "sortedNums"
            sortedNums.insert(sortedNums.begin() + pos, nums[i]);
        }
        return res;
    }
};