剑指offer.51 --- 数组中的逆序对

数组中的逆序对

题目链接：https://leetcode-cn.com/problems/shu-zu-zhong-de-ni-xu-dui-lcof/

首先想到的方法是暴力解法，用2个for循环遍历数组，比较前一个和后一个的大小，时间复杂度为O(N^2),这里给的数字范围很大，这个方法明显就不行了。

class Solution {
     
public:
    int reversePairs(vector<int>& nums) {
       int n = nums.size();
       int count = 0;
         for(int i = 0; i < n;++i){
             for(int j = i + 1; j < n;++j){
                 if(nums[i] > nums[j]){
                     count++;
                 }
             }
         }
     return count;
 }
};

因为是前面的数大于后面的数，我们可以想到归并排序，归并排序的思想：

• 确定一个中间的位置
• 递归排序左区间和右区间
• 归并，将2个有序的合并到临时数组，最后还原数组

归并的过程如下：

利用归并时求出逆序对的个数：
如图：

动图如下：

整体的流程：

• 终止条件： 当 l ≥ r 时，代表子数组长度为 1 ，此时终止划分；
• 递归划分： 计算数组中点 mid ，递归划分左子数组 merge_sort(l, mid) 和右子数组merge_sort(mid+1, r) ；
• 逆序对总数 = merge_sort(l, mid) + merge_sort(mid+1, r) ；
• 循环合并： 设置双指针 i , j分别指向左 / 右子数组的首元素,循环结束条件：i <= mid && j <= r,比较2个数的大小，将左子数组中的数直接临时数组，将右子数组中放入临时数组时，逆序对的个数为: mid - i + 1
• 当其中一个子数组放完后将另一个没放完的直接放入临时数组
• 将临时数组拷回原数组
• 返回值： 返回直至目前的逆序对总数 res ；

代码如下：

class Solution {
     
public:
  int reversePairs(vector<int>& nums)
  {
    vector<int> tmp(nums.size());
      
    return merge_sort(0, nums.size()-1,nums,tmp);    
  }
 int merge_sort(int l,int r,vector<int>& q,vector<int>& tmp)
 {
    if(l >= r) return 0; //当区间只剩1个元素时结束
    int mid = (l + r ) / 2;
    int  res = merge_sort(l,mid,q,tmp) + merge_sort(mid+1,r,q,tmp);
    int k = 0,i = l,j = mid + 1;
    while(i <= mid && j <=r)
    {
       if(q[i] <= q[j])
       {
          tmp[k++] = q[i++];  //左区间小的直接放入临时数组
       }
       else{
         res += mid - i + 1; //比如 456｜123，right 的 1 进 tmp
         tmp[k++] = q[j++];  //那么就产生了 3 个逆序对，2进 tmp，产生三个逆序对，。。。
                            // 规律就是当前 left 中剩余元素的数目，可以用 mid - i + 1 来计算
        }
      }
   while(i <= mid) tmp[k++] = q[i++]; //当其中1个结束时将令1个剩下的数放入临时数组
   while(j <= r)  tmp[k++] = q[j++];
   
   for(int i = l,j = 0;i <= r;++i,++j) q[i] = tmp[j];//将临时数组拷回原数组
      return res;
    }
};

复杂度：

• 时间复杂度：O(N log N)，这里 N 是数组的长度。复杂度是归并排序的时间复杂度，直接看递归树的结点个数或者使用主定理分析，归并的回收每一步计算逆序对的个数是 O(1) 的；
• 空间复杂度：O(N)。使用了额外的数组

这题主要考察的是归并排序。在排序中我们要熟练使用快排、堆排、快排，多加练习。