大小为a
的数组n
中的反转称为成对(i,j)
,其i<j
和a[i]>a[j]
。我正在尝试用C ++实现一个函数,它计算给定数组中的反转次数。我遵循分而治之的方法,它只修改了合并排序算法,并在O(n log n)时间内运行。到目前为止,这是我的代码:
long long int glob;
template< class T >
long long int merge( T *arr, int beg, int mid, int end ) {
queue< int > left;
queue< int > right;
for( int i=beg; i<=mid; ++i ) {
left.push( arr[i] );
}
for( int i=mid+1; i<=end; ++i ) {
right.push( arr[i] );
}
int index=beg;
int ret=0;
while( !left.empty() && !right.empty() ) {
if( left.front() < right.front() ) {
arr[index++] = left.front();
left.pop();
} else {
arr[index++] = right.front();
right.pop();
ret+=left.size();
}
}
while( !left.empty() ) { arr[index++]=left.front();left.pop(); }
while( !right.empty() ) { arr[index++]=right.front();right.pop(); }
return ret;
}
template< class T >
void mergesortInvCount( T *arr, int beg, int end ) {
if( beg < end ) {
int mid = (int)((beg+end)/2);
mergesortInvCount( arr, beg, mid );
mergesortInvCount( arr, mid+1, end );
glob += merge( arr, beg, mid, end );
}
}
对于某些测试用例,它会产生正确的结果,但对于其他一些测试用例,它会给出错误的输出。我是否错误地理解了算法,或者我在实现时犯了错误?有人能帮帮我吗?提前致谢。
Test case: 2 1 3 1 2
Correct: 4
Mine: 6
答案 0 :(得分:2)
我没有查看代码中的所有步骤,但是您的行
if( left.front() < right.front() )
向我建议,在else分支中“ret”不仅在a(j)&gt; a(i)时增加,而且在它们相等时增加,这与你对案例的描述不符。所以也许你应该尝试将上面引用的行改为:
if( left.front() <= right.front() )
此致
答案 1 :(得分:1)
测试
if( left.front() < right.front() )
应为<=
。你现在正在从左半部分之前的右半部分移动相同的值,计算不是的倒数(每个这样的事件计算左边的相同项目的虚假反转数)。在您的示例中,您必须复制对,每个对创建一个幻像反转。