牛客网 “数组中的逆序对” 题目:
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出(即输出P%1000000007 )。
输入描述: 题目保证输入的数组中没有的相同的数字
对于 %50 的数据, size<=104
对于 %75 的数据, size<=105
对于 %100 的数据, size<=2*105
样例输入:1,2,3,4,5,6,7,0
样例输出:7
用归并排序的思想统计逆序对:先把数组分割成子数组,再子数组合并的过程中统计逆序对的数目。统计逆序对时,先统计子数组内部的逆序对的数目,再统计相邻两个有序子数组的逆序对数目。
int cnt = 0;
int InversePairs(vector<int> data) {
MergeSort(data,0,data.size()-1);
return cnt;
}
/* 将长度为 n 的输入序列分成两个长度为 n/2 的子序列 */
void MergeSort(vector<int>& a, int l, int r) {
if (l >= r) return;
const int mid = l + (r - l) / 2; // l + ((r - l) >> 1); !!!
// 递归拆分
MergeSort(a, l, mid);
MergeSort(a, mid + 1, r);
// 递归合并
Merge(a, l, mid, r);
}
void Merge(vector<int>& a, int l, int m, int r) {
vector<int> t;
int i = l; /* i指向第一个子表 */
int j = m + 1;/* j指向第二个子表 */
/* 两个子表都不为空时 */
while(i <= m && j <= r) {
/* 取关键字小的元素转移至临时数组 */
if (a[i] > a[j]) {
t.push_back(a[j++]);
cnt = (cnt + m - i + 1) % 1000000007;
}
else
t.push_back(a[i++]);
}
while(i <= m) t.push_back(a[i++]);/* 将非空的输入区间转移至输出区间 */
while(j <= r) t.push_back(a[j++]);
for (i = 0; i < t.size(); i++)
a[l + i] = t[i];/* 归并完成后将结果复制到原输入数组 */
}