首页 > 其他 > 详细

BIT-Reverse Pairs

时间:2019-12-17 11:47:47      阅读:105      评论:0      收藏:0      [点我收藏+]

2019-12-17 11:07:02

问题描述

技术分享图片

 

问题求解

本题可以看作是逆序数问题的强化版本,需要注意的是num[i] > 2 * num[j],这里有0和负数的情况。

    public int reversePairs(int[] nums) {
        int res = 0;
        int n = nums.length;
        int[] nums_copy = Arrays.copyOf(nums, n);
        TreeMap<Integer, Integer> map = new TreeMap<>();
        Arrays.sort(nums_copy);
        int rank = 0;
        for (int i = 0; i < n; i++) {
            if (i == 0 || nums_copy[i] != nums_copy[i - 1]) 
                map.put(nums_copy[i], ++rank);
        }
        int[] bit = new int[map.size() + 1];
        for (int i = n - 1; i >= 0; i--) {
            int num = nums[i] % 2 == 0 ? nums[i] / 2 - 1 : (nums[i] - 1) / 2;
            Integer key = map.floorKey(num);
            if (key != null) res += query(bit, map.get(key));
            update(bit, map.get(nums[i]));
        }
        return res;
    }
    
    private void update(int[] bit, int idx) {
        for (int i = idx; i < bit.length; i += i & -i) {
            bit[i] += 1;
        }
    }
    
    private int query(int[] bit, int idx) {
        int res = 0;
        for (int i = idx; i > 0; i -= i & -i) {
            res += bit[i];
        }
        return res;
    }

  

 

BIT-Reverse Pairs

原文:https://www.cnblogs.com/hyserendipity/p/12053332.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!