数组中的逆序对
1.题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
输入描述:
题目保证输入的数组中没有的相同的数字
数据范围:
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5
示例1
2.思路
所谓逆序对,就是指在数组给定的顺序中,如果排在前面的数字的值比排在它后面的数字的值大,那么这两个数字就构成一个逆序对。
我们以数组{7,5,6,4}为例来分析统计逆序对的过程。每次扫描到一个数字的时候,我们不拿它和后面的每一个数字作比较,否则时间复杂
O
(
n
2
)
O(n^2)
O(n2)。因此我们可以考虑先比较两个相邻的数字。如下图所示:
在上图(a)和(b)中,我们先把数组分解成两个长度为2的子数组,再把这两个子数组分别拆成两个长度为1的子数组。接下来一边合并相邻的子数组,一边统计逆序对的数目。在第一对长度为1的子数组{7}、{5}中7大于5,因此(7,5)组成一个逆序对。同样在第二对长度为1的子数组{6}、{4}中也有逆序对(6,4)。由于我们已经统计了这两对子数组内部的逆序对,因此需要把这两对子数组排序如上图(c)所示, 以免在以后的统计过程中再重复统计。接下来我们统计两个长度为2的子数组子数组之间的逆序对。合并子数组并统计逆序对的过程如下图所示。
我们先用两个指针分别指向两个子数组的末尾,并每次比较两个指针指向的数字。如果第一个子数组中的数字大于第二个数组中的数字,则构成逆序对,并且逆序对的数目等于第二个子数组中剩余数字的个数,如(a)和(c)所示。如果第一个数组的数字小于或等于第二个数组中的数字,则不构成逆序对,(b)所示。每一次比较的时候,我们都把较大的数字从后面往前复制到一个辅助数组中,确保辅助数组(记为copy) 中的数字是递增排序的。在把较大的数字复制到辅助数组之后,把对应的指针向前移动一位,接下来进行下一轮比较。
过程:先把数组分割成子数组,先统计出子数组内部的逆序对的数目,然后再统计出两个相邻子数组之间的逆序对的数目。在统计逆序对的过程中,还需要对数组进行排序。如果对排序算法很熟悉,我们不难发现这个过程实际上就是归并排序。
3.代码
class Solution {
public:
int InversePairs(vector<int> data) {
if(data.empty() || data.size() < 2){
return 0;
}
vector<int> copy(data.size(),0);
long long count = mergeSort(data,0,data.size()-1,copy);
return count % 1000000007;
}
long long mergeSort(vector<int>& data, int begin,int end,vector<int>& copy){
if(begin >= end){
return 0;
}
int mid = (begin + end) / 2;
long long lcount = mergeSort(data,begin,mid,copy);
long long rcount = mergeSort(data,mid+1,end,copy);
int i = mid;
int j = end;
int copyIndex = end;
long long count = 0;//记录逆序对的数量
while(i >= begin && j >= mid + 1){
if(data[i] > data[j]){
copy[copyIndex--] = data[i--];
count += (j - mid);
}
else{
copy[copyIndex--] = data[j--];
}
}
//将子数组剩余部分一次写入归并后的存储数组
while(i >= begin){
copy[copyIndex--] = data[i--];
}
while(j >= mid + 1){
copy[copyIndex--] = data[j--];
}
//将本次两个子数组的合并写入原数组中
for(int i = begin; i <= end; i++){
data[i] = copy[i];
}
return lcount + rcount + count;
}
};
4.复杂度分析
时间复杂度:O(nlogn)
空间复杂度:O(n)