[剑指Offer] 35.数组中的逆序对

题目描述

在数组中的两个数字，如果前面一个数字大于后面的数字，则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007

【思路】看到这样的题目，最简单的想法就是遍历每一个元素，让其与后面的元素对比，如果大于则count++，但是这样的时间复杂度是o(n²)，根据题目给出的数据量，很明显超时。因此想到了用归并排序的思想。

 class Solution {
 public:
     void MergeCount(vector<int>& data, int begin, int mid, int end, int& count)
     {
         int i = begin, j = mid;
         int temp[end - begin],  k = ;
         while(i < mid && j < end)
         {
             if(data[i] < data[j])
             {
                 temp[k++] = data[i++];
             }
             else
             {
                 count += mid - i;
                 if(count > ) count %= ;
                 temp[k++] = data[j++];
             }
         }
         while(i < mid) temp[k++] = data[i++];
         while(j < end) temp[k++] = data[j++];
         for(int s = ; s < end - begin; ++s) data[s + begin] = temp[s];
     }
     void MergeSort(vector<int>& data, int begin, int end, int& count)
     {
         if(begin +  == end) return;
         int mid = (begin + end) / ;
         MergeSort(data, begin, mid,count);
         MergeSort(data, mid, end, count);
         MergeCount(data, begin, mid, end, count);
     }
     int InversePairs(vector<int> data)
     {
         int n = static_cast<int>(data.size());
         if(n ==  || n == ) return ;
         int count = ;
         MergeSort(data,, n, count);
         return count;
     }
 };