汇芳书院

专注计算机视觉、机器学习、分布式计算等领域, 兼聊投资、写作、生活

0%

剑指 Offer 51. 数组中的逆序对

发表于 更新于 分类于 Valine：
本文字数： 2.7k 阅读时长 ≈ 2 分钟

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组，求出这个数组中的逆序对的总数。

 

示例 1:

输入: [7,5,6,4]
输出: 5

暴力解法

时间复杂度O(n^2), 无法通过OJ

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution {
public:
    int reversePairs(vector<int>& nums) {
        int n = nums.size();
        int ans = 0;
        for (int i = 0; i < n; i++) {
            for (int j = i+1; j < n; j++) {
                if (nums[j] < nums[i]) {
                    ans++;
                }
            }
        }
        return ans;

    }
};

归并排序

两个有序数组merge的时候, 当将要选右边数组合并到新数组时，被选中的这个数比左半数组中的数都要小，
此时逆序对数增加为左边数组中的长度。

时间复杂度O(nlogn)
空间复杂度O(n)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
class Solution {
    int ans = 0;
public:
    int reversePairs(vector<int>& nums) {
        int n = nums.size();
        mergeSort(nums, 0, n-1);
        return ans;
    }

    void mergeSort(vector<int>& nums, int left, int right) {
        int n = nums.size();
        if (left >= right) {
            return;
        }
        int mid = left + (right-left)/2;
        mergeSort(nums, left, mid);
        mergeSort(nums, mid+1, right);
        //如果拆分后的两部分已经有序，则不用merge
        if (nums[mid] <= nums[mid+1]) {
            return;
        }
        merge(nums, left, mid, right);
    }

    //[left, mid], [mid+1, right]
    void merge(vector<int>& nums, int left, int mid, int right) {
        vector<int> tmp;
        int i = left, j = mid+1;
        while (i <= mid && j <= right) {
            if (nums[i] <= nums[j]) {
                tmp.emplace_back(nums[i]);
                i++;
            }else{
                ans += mid-i+1;
                tmp.emplace_back(nums[j]);
                j++;
            }
        }
        while (i <= mid) {
            tmp.emplace_back(nums[i]);
            i++;
        }
        while (j <= right) {
            tmp.emplace_back(nums[j]);
            j++;
        }
        for (int i = left, j=0; i <= right; i++,j++) {
            nums[i] = tmp[j];
        }
    }

    
};

临时数组仅申请一次，降低中间申请销毁的时间

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
class Solution {
    int ans = 0;
public:
    int reversePairs(vector<int>& nums) {
        int n = nums.size();
        vector<int> tmp(n);
        mergeSort(nums, tmp, 0, n-1);
        return ans;
    }

    void mergeSort(vector<int>& nums, vector<int>& tmp, int left, int right) {
        int n = nums.size();
        if (left >= right) {
            return;
        }
        int mid = left + (right-left)/2;
        mergeSort(nums, tmp, left, mid);
        mergeSort(nums, tmp, mid+1, right);
        //如果拆分后的两部分已经有序，则不用merge
        if (nums[mid] <= nums[mid+1]) {
            return;
        }
        merge(nums, tmp, left, mid, right);
    }

    //[left, mid], [mid+1, right]
    void merge(vector<int>& nums, vector<int>& tmp, int left, int mid, int right) {
        int i = left, j = mid+1, k=left;
        while (i <= mid && j <= right) {
            if (nums[i] <= nums[j]) {
                tmp[k++] = nums[i++];
            }else{
                ans += mid-i+1;
                tmp[k++] = nums[j++];
            }
        }
        while (i <= mid) {
            tmp[k++] = nums[i++];
        }
        while (j <= right) {
            tmp[k++] = nums[j++];
        }
        copy(tmp.begin()+left, tmp.begin()+right+1, nums.begin()+left);
        // for (int i = left,k=left; i <= right; i++,k++) {
        //     nums[i] = tmp[k];
        // }
    }
    
};

离散化树状数组

TODO

坚持原创分享，您的支持将鼓励我继续创作

欢迎关注我的其它发布渠道

------------- 本文结束,感谢阅读 如有问题可留言交流 -------------