在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数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
/*归并排序的改进,把数据分成前后两个数组(递归分到每个数组仅有一个数据项),
合并数组,合并时,出现前面的数组值array[i]大于后面数组值array[j]时;则前面
数组array[i]~array[mid]都是大于array[j]的,count
+= mid+1 - i
参考剑指Offer,但是感觉剑指Offer归并过程少了一步拷贝过程。
还有就是测试用例输出结果比较大,对每次返回的count
mod(1000000007)求余
*/
public
class
Solution {
public
int
InversePairs(
int
[] array) {
if
(array==
null
||array.length==
0
)
{
return
0
;
}
int
[]
copy =
new
int
[array.length];
for
(
int
i=
0
;i<array.length;i++)
{
copy[i]
= array[i];
}
int
count = InversePairsCore(array,copy,
0
,array.length-
1
);
//数值过大求余
return
count;
}
private
int
InversePairsCore(
int
[]
array,
int
[]
copy,
int
low,
int
high)
{
if
(low==high)
{
return
0
;
}
int
mid = (low+high)>>
1
;
int
leftCount = InversePairsCore(array,copy,low,mid)%
1000000007
;
int
rightCount = InversePairsCore(array,copy,mid+
1
,high)%
1000000007
;
int
count =
0
;
int
i=mid;
int
j=high;
int
locCopy = high;
while
(i>=low&&j>mid)
{
if
(array[i]>array[j])
{
count
+= j-mid;
copy[locCopy--]
= array[i--];
if
(count>=
1000000007
)
//数值过大求余
{
count%=
1000000007
;
}
}
else
{
copy[locCopy--]
= array[j--];
}
}
for
(;i>=low;i--)
{
copy[locCopy--]=array[i];
}
for
(;j>mid;j--)
{
copy[locCopy--]=array[j];
}
for
(
int
s=low;s<=high;s++)
{
array[s]
= copy[s];
}
return
(leftCount+rightCount+count)%
1000000007
;
}
}
参考《剑指offer》用归并排序的思想, 时间复杂度O(nlogn)
代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
int
InversePairs(vector< int >
data) { int
length = data.size(); return
mergeSort(data, 0, length-1); } int
mergeSort(vector< int >&
data, int
start, int
end) { //
递归终止条件 if (start
>= end) { return
0; } //
递归 int
mid = (start + end) / 2; int
leftCounts = mergeSort(data, start, mid); int
rightCounts = mergeSort(data, mid+1, end); //
归并排序,并计算本次逆序对数 vector< int >
copy(data); //
数组副本,用于归并排序 int
foreIdx = mid; //
前半部分的指标 int
backIdx = end; //
后半部分的指标 int
counts = 0; //
记录本次逆序对数 int
idxCopy = end; //
辅助数组的下标 while (foreIdx>=start
&& backIdx >= mid+1) { if (data[foreIdx]
> data[backIdx]) { copy[idxCopy--]
= data[foreIdx--]; counts
+= backIdx - mid; } else
{ copy[idxCopy--]
= data[backIdx--]; } } while (foreIdx
>= start) { copy[idxCopy--]
= data[foreIdx--]; } while (backIdx
>= mid+1) { copy[idxCopy--]
= data[backIdx--]; } for ( int
i=start; i<=end; i++) { data[i]
= copy[i]; } return
(leftCounts+rightCounts+counts); }
|