知识点:归并排序
一、题目
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
二、思路
1、暴力解法,顺序遍历整个数组,每遍历到一个数字的时候,逐个比较该数字和它后面数字的大小。如果后面的数字比它小,则这两个数字就组成一个逆序对。由于数组较大,时间复杂度限制不通过。
class Solution {
public:
int InversePairs(vector<int> data) {
long inPairs = 0;
for(int i=0; i<data.size(); i++){
for(int j=i+1; j<data.size(); j++){
if(data[i] > data[j]){
inPairs++;
}
}
}
return (inPairs % 1000000007);
}
};
2、分治思想,采用归并排序的思路处理,先分后治。以数组data={5,3,2,6}为例
上述(3)步中表述有误,应该是将长度为1的数组进行合并、排序、并统计逆序对。
C++实现:
class Solution {
public:
long cnt = 0;
//合并子数组并一边统计逆序对数目
void merge(vector<int> &data,int start,int mid,int end){
vector<int> tmp;//存储当前[start,end]段排序的结果,然后再赋值给data
int i = start,j = mid+1;
while(i<=mid && j<=end){
if(data[i] > data[j]){
cnt = (cnt+mid-i+1)%1000000007;
tmp.push_back(data[j++]);
}
else{
tmp.push_back(data[i++]);
}
}
while(i<=mid){
tmp.push_back(data[i++]);
}
while(j<=end){
tmp.push_back(data[j++]);
}
for(int i=0;i<tmp.size();i++){
data[start + i] = tmp[i];
}
}
//拆分数组至字数组长度为1,再进行合并并统计逆序对的数目
void divide(vector<int> &data, int start, int end){
if(start < end){
int mid = (start + end) >> 1;
//递归,分别拆分子左区间和右子区间
divide(data,start,mid);
divide(data,mid+1,end);
//合并子区间并统计逆序对数目
merge(data,start,mid,end);
}
}
int InversePairs(vector<int> data) {//思考形参为啥不是vector<int> &data?
if(data.size() == 0){
return 0;
}
divide(data,0,data.size()-1);
return cnt;
}
};
理解过程: