数组中的逆序对 - 我们不一样
看到一道很有趣的题,别人基本上都是用“归并排序”,我就换了一种思路。但是让你沮丧的是,网上运行结果,提示“运行超时”,也没说那个测试用例超时,也看不到测试用例。我个人认为我的算法没有问题,时间复杂度是O(n*logn)。特意记录一下,有哪位大牛能找出我的错误,请留言。
原题链接:https://www.nowcoder.com/questionTerminal/96bd6684e04a44eb80e6a68efc0ec6c5
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
输入描述:
题目保证输入的数组中没有的相同的数字
数据范围:
对于%50的数据,size<=10^4
对于%75的数据,size<=10^5
对于%100的数据,size<=2*10^5
示例1
输入
1,2,3,4,5,6,7,0
输出
7
#include <iostream>
#include <algorithm>
#include <vector>
//tdw
using namespace std;
class Solution {
public:
int InversePairs(vector<int> data) {
vector<int> vSortNum;//已排好顺序
vSortNum.reserve(data.size());
int mid = 0;
int beg = 0;
int end = 0;
int sum = 0;//逆序对的数量
if (data.size() <= 1)
return 0;
vSortNum.push_back(data[0]);
for (int i = 1; i < data.size(); ++i)//扫描一遍,如果“data[i]”插入的位置是"i",则逆序对为0,否者逆序对+(size()-实际位置)
{
int num = data[i];
beg = 0;
end = vSortNum.size() - 1;
if (vSortNum[end] < num)
{//插入最后一个
vSortNum.push_back(num);
continue;
}
if (vSortNum[beg] > num)
{//插入到第一个
sum += vSortNum.size();//计算插入的位置,与实际位置的差值
if (sum > 1000000007)
sum %= 1000000007;
vSortNum.insert(vSortNum.begin(), num);
continue;
}
while (true)//折半查找要插入的位置
{//已保证不存在相同的两个数
mid = (beg + end) / 2;
if (vSortNum[mid] > num)
{
end = mid - 1;
}
else if (vSortNum[mid] < num)
{
beg = mid + 1;
}
if (beg > end)//已找到相应插入的位置位置beg
{
sum += vSortNum.size() - beg;//计算插入的位置,与实际位置的差值
if (sum > 1000000007)
sum %= 1000000007;
vSortNum.insert(vSortNum.begin() + beg, num);
break;
}
}
}
return sum;
}
};
int main()
{
Solution s;
vector<int> v = { 1,2,3,7,5,6,4,0 };
int sum = s.InversePairs(v);
cout << sum;
return 0;
}