(C++)剑指offer-35:数组中的逆序对
如果采用暴力枚举的方法,顺序扫描整个数组,每扫描到一个数的时候,逐个比较该数字和它后面的数字的大小。如果后面的数字比它小,则这两个数字就组成了一个逆序对。利用这种暴力冒泡排序的方法,时间复杂度为O(n2)。因此考虑时间复杂度更低的归并排序的方法,把数组分成两部分,分别递归计算左右两部分数组中存在的逆序对数merge1和merge2,再考虑两数组之间存在的逆序对,定义两个指针i,j,分别从数组的l和mid+1开始右移动,如果发现a[i] > a[j],j右移一位并且res加上mid-i+1个,否则i右移一位,直到两指针有一个遍历结束。最后整体的逆序对数就为左右两数组内的逆序对加上两数组之间存在的逆序对。
采用归并方法使得时间复杂度降至O(nlogn)。牛客需考虑size大的样例情况,具体代码如下:
class Solution {
public:
int merge(vector<int> &data, int l, int r){
if(l >= r) return 0;
int mid = l + r >> 1;
int res = merge(data, l, mid) + merge(data, mid + 1, r);
vector<int> temp;
int i = l, j = mid + 1;
while(i <= mid && j <= r){
if(data[i] <= data[j]) temp.push_back(data[i++]);
else{
temp.push_back(data[j++]);
res += mid - i + 1;
res %= 1000000007;
}
}
while(i <= mid) temp.push_back(data[i++]);
while(j <= r) temp.push_back(data[j++]);
i = l;
for(auto x : temp) data[i++] = x; //注意将原先数组还原
return res;
}
int InversePairs(vector<int>& data) {
return merge(data, 0, data.size() - 1);
}
};