题目要求
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P
解题分析
最容易想到的方法就是对每个元素都遍历其后边的元素,这样就能找到所有的逆序对,但是这样的办法是平方级的时间复杂度不适合我们做面试。
所以我们不把当前元素和其后边所有的元素比较,而是只比较相邻的两个元素,把数组一分为二,二分为四,直到每个子数组只有一个元素,然后我们可以根据相邻的两个数组进行比较,如果前一个数组中的元素值比后一个大那么就是一个逆序对,否则就不是。之后合并刚刚比较完成的数组,并对数组进行排序,避免在下一次的计算中重复记录。 特别地,逆序对的个数等于第二个子数组中剩余数字的个数。
合并的过程有点像归并排序,另外需要注意的是,由于涉及到数字相加,很可能超过整数的表示范围,所以是一个隐含的大数问题,所以要解决的话:一是再设定计数次数的时候统一成long long型,或者对计数次数进行取余操作,根据题目要求本解答运用第二种方法。
主要代码 c++
class Solution {
public:
int InversePairs(vector<int> data) {
int size = data.size();
if(size == 0) return 0;
vector<int>copy(size);
for(int i=0;i<size;++i)
copy[i] = data[i]; // 辅助数组
int count = MergeSort(data,copy,0,size-1); //定义int 注意大数问题 相加之后可能会超过整数范围
return count;
}
int MergeSort(vector<int>&data, vector<int>©, int start, int end)
{
if(start == end)
{
return 0;
}
int mid = (start+end)/2;
int LeftCount = MergeSort(copy,data,start,mid); // 划分左右数组
int RightCount = MergeSort(copy,data,mid+1,end);
//定义两个指针分别指向前/后连个数组的最后一个元素
int FirstInd = mid;
int SecondInd = end;
//定义辅助数组的指针指向
int CopyInd = end;
int counts = 0;
while(FirstInd>=start && SecondInd>=mid+1)
{
if(data[FirstInd]>data[SecondInd])
{
copy[CopyInd--] = data[FirstInd--];
counts += SecondInd - mid;
if(counts > 1000000007) // 防止相加之后产生大数问题
counts %= 1000000007;
}
else
copy[CopyInd--] = data[SecondInd--];
}
for(; FirstInd>=start; --FirstInd)
copy[CopyInd--] = data[FirstInd];
for(; SecondInd>=mid+1; --SecondInd)
copy[CopyInd--] = data[SecondInd];
return (LeftCount + RightCount + counts)%1000000007;// 防止相加之后产生大数问题
}
};