题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数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
算法分析
思路一:顺序扫描整个数组,每扫描到一个数字的时候,逐个比较数字和它后面的数字的大小,如果后面的数字比它小,则这两个数字就组成逆序对。假设数组中含有n个数字。由于每个数字都要和O(n)个数字比较,所以算法算法的时间复杂度为O(n^2)
思路二:利用归并排序的思想,将数组每次折半分开,并统计其中的逆序对,然后再排序合并,缺点是会使用辅助数组。这样做相当于在归并排序中加入了比较记录信息。同时消耗了一个长度为n的辅助数组,相当于用O(n)的空间消耗换来了时间效率的提升,典型的空间换时间的做法。
代码
class Solution {
public:
int InversePairs(vector<int> data) {
return (int)dfs(data, 0, data.size());
}
long long dfs(vector<int> &data, int l, int r) {
if (r - l == 1)
return 0;
const long long mod = 1000000007;
int mid = (l + r) >> 1, ret = (dfs(data, l, mid) % mod + dfs(data, mid, r) % mod) % mod;
vector<int> arr;
int left = l, right = mid;
for (; left < mid && right < r; )
if (data[left] > data[right])
arr.push_back(data[right++]), ret = (ret + mid - left) % mod;
else
arr.push_back(data[left++]);
for (; left < mid; arr.push_back(data[left++]));
for (; right < r; arr.push_back(data[right++]));
for (int i = 0; l < r; data[l++] = arr[i++]);
return ret % mod;
}
};
这里有一份注释很详细的代码可以参考一下:
链接:https://www.nowcoder.com/questionTerminal/96bd6684e04a44eb80e6a68efc0ec6c5
来源:牛客网
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--];
//统计长度
count+=j-start-length;
if(count>=1000000007)//数值过大求余
count%=1000000007;
}
else
{
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;
}
};