LetCode 493. 翻转对(归并排序)

本题思路:在递归函数中递归后,先不合并,首先处理逆序数,然后再合并。这样需要再写一个合并函数。这样的好处是在处理计数的时候不需要考虑谁先被写入copy数组。使得复杂度降低。

归并排序看这个

第一种:二分法,妥妥垫底 1500ms 差点超时

class Solution {
public:
    int reversePairs(vector<int>& nums) {
        long cnt = 0;
        vector<long> trans;
        if (nums.size() == 0)
            return cnt;
        trans.push_back((long)2 * nums[nums.size() - 1]);
        for (int i = nums.size() - 2; i >= 0; i--) {
            auto it = lower_bound(trans.begin(), trans.end(), nums[i]);
            cnt += it - trans.begin();
            it = lower_bound(trans.begin(), trans.end(), (long)2 * nums[i]);
            trans.insert(it, (long)2 * nums[i]);
        }
        return cnt;
    }
};
第二种:归并排序,160ms
class Solution {
public:
    int reversePairs(vector<int>& nums) {
        int n = nums.size();
        if (n <= 0)
            return 0;
        ans = new double[n];
        res = 0;
        sort(nums, 0, n - 1);
        return res;
    }
    double* ans;
    int res;
    
    void sort(vector<int>& nums, int l, int r){
        if (r <= l)
            return;
        int m = (l + r) >> 1;
        sort(nums, l, m);
        sort(nums, m + 1, r);
        int i = l, j = m + 1;
        while(j <= r){
            while(i <= m && (double)nums[i] / 2 <= nums[j]) 
                i++;
            res += m - i + 1;
            j++;
        }
        merge(nums, l, m, r);
    }
    
    void merge(vector<int>& nums, int l, int m, int r){
        for (int i = l; i <= r; i++)
            ans[i] = nums[i];
        
        int i = l, j = m + 1;
        int k = l,n = r;
        while(i <= m && j <= n){
            if (ans[i] <= ans[j])
                nums[k++] = ans[i++];
            else
                nums[k++] = ans[j++];
        }
        
        while(i <= m)
            nums[k++] = ans[i++];
        while(j <= n)
            nums[k++] = ans[j++];
    }
};

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值