数组中的逆序对
题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
输入描述:
题目保证输入的数组中没有的相同的数字
数据范围:
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5
示例输入
1,2,3,4,5,6,7,0
输出
7
解题思路
- 如果用每个数和他后面的数比较,统计逆序对数,那么时间复杂度是O(N^2)
- 采用类似归并排序的思想,以数组{7,5,6,4}为例。每次扫描到一个数字的时候,不拿它和后面的每一个数字作比较,否则时间复杂度就是O(n^2),考虑先比较两个相邻的数字
- (a) 把长度为4的数组分解成两个长度为2的子数组
- (b) 把长度为2的数组分解成两个成都为1的子数组
- (c ) 把长度为1的子数组合并、排序并统计逆序对
- (d) 把长度为2的子数组合并、排序,并统计逆序对
- 在上图(a)和(b)中,先把数组分解成两个长度为2的子数组,再把这两个子数组分别拆成两个长度为1的子数组。接下来一边合并相邻的子数组,一边统计逆序对的数目
- 在第一对长度为1的子数组{7}、{5}中7大于5,因此(7,5)组成一个逆序对。同样在第二对长度为1的子数组{6}、{4}中也有逆序对(6,4)。由于已经统计了这两对子数组内部的逆序对,因此需要把这两对子数组排序 如上图(c)所示, 以免在以后的统计过程中再重复统计
- 接下来统计两个长度为2的子数组之间的逆序对。合并子数组并统计逆序对的过程如下图所示:
- 先用两个指针分别指向两个子数组的末尾,并每次比较两个指针指向的数字。如果第一个子数组中的数字大于第二个数组中的数字,则构成逆序对,并且逆序对的数目等于第二个子数组中剩余数字的个数,如上图(a)和(c)所示。如果第一个数组的数字小于或等于第二个数组中的数字,则不构成逆序对,如图中b所示
- 每一次比较的时候,我们都把较大的数字从后面往前复制到一个辅助数组中,确保辅助数组(记为copy) 中的数字是递增排序的。在把较大的数字复制到辅助数组之后,把对应的指针向前移动一位,接下来进行下一轮比较
- 过程总结:先把数组分割成子数组,统计出子数组内部的逆序对的数目,然后再统计出两个相邻子数组之间的逆序对的数目。在统计逆序对的过程中,还需要对数组进行排序,这就类似于排序算法中的归并排序
class Solution {
public:
int InversePairs(vector<int> data) {
if(data.size()<=1) return 0;//如果少于等于1个元素,直接返回0
int* copy=new int[data.size()];//辅助数组
//初始化辅助数组,作为存放临时排序的结果,最后要将排序的结果复制到原数组中
for(unsigned int i=0;i<data.size();i++)
copy[i]=0;
//调用递归函数求解逆序对数
int count=InversePairCore(data,copy,0,data.size()-1);
delete[] copy;//删除临时数组
return count;
}
//统计逆序对的函数
int InversePairCore(vector<int>& data,int*& copy,int start,int end)
{
if(start==end){
copy[start]=data[start];
return 0;
}
//将数组拆分成两部分
int length=(end-start)/2;//这里使用的是下标法,下面要用来计算逆序个数;也可以直接使用mid=(start+end)/2
//分别计算左边部分和右边部分,递归的将数组两两拆分
int left=InversePairCore(data,copy,start,start+length)%1000000007;
int right=InversePairCore(data,copy,start+length+1,end)%1000000007;
//进行逆序计算
int i=start+length;//前一个数组的最后一个下标
int j=end;//后一个数组的下标
int index=end;//辅助数组下标,从最后一个算起
int count=0;
while(i>=start && j>=start+length+1){
if(data[i]>data[j]){//有逆序对
copy[index--]=data[i--];//大的复制到辅助数组
//统计长度,j及其之前的元素都比data[i]小
count+=j-start-length;//j-(start+length)
if(count>=1000000007)//数值过大,求余
count%=1000000007;
}
else{//不是逆序对,正常复制,下标前移一位
copy[index--]=data[j--];
}
}
//某一半边剩下的直接复制,跟合并有序数组中一样,下面两个while只会执行一个
while(i>=start){
copy[index--]=data[i--];
}
while(j>=start+length+1){
copy[index--]=data[j--];
}
/*
for(;i>=start;--i){
copy[index--]=data[i];
}
for(;j>=start+length+1;--j){
copy[index--]=data[j];
}*/
//将统计逆序对并排序后的数组复制回去
for(int i=start;i<=end;i++) {
data[i] = copy[i];
}
//返回最终的结果
return (count+left+right)%1000000007;
}
};