题目
http://poj.org/problem?id=1804
题意:求数组中的逆序对个数。逆序对指的是满足ai > aj的数对(ai, aj)。
变形:http://acm.ecnu.edu.cn/problem/3234/
给定一个无序序列,每次只能交换相邻两个数字,求最小的交换次数。(算法是每次都将最大的交换到最后,即求逆序对个数)
解题思路
在归并排序的合并过程中,加入统计逆序对的语句即可。
归并排序利用了分治的思想,将数列a[l, r]分成两半a[l, mid]和a[mid+1, r],分别进行归并排序,然后再将这两半合并(merge)起来。
在合并的过程中(设l<=i<=mid,mid+1<=j<=h),当a[i]<=a[j]时,并不产生逆序数;当a[i]>a[j]时,在前半部分中比a[i]大的数都比a[j]大,将a[j]放在a[i]前面时,逆序数要加上mid+1-i。因此,可以在归并排序中的合并过程中计算逆序数。
合并操作时,i是左半部分数组的下标,j是右半部分数组的下标,k是存放合并结果的临时数组的下标,一定要注意下标的自增,以及最后两个半边数组比较跳出以后,多余部分的抄回。
AC代码
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1005;
int a[maxn], tmp[maxn], ans;
void Merge(int l, int mid, int r) //归并过程
{
int i = l, j = mid + 1;
int k = l; //临时数组tmp的下标
while (i <= mid && j <= r)
{
if (a[i] > a[j]) //a[i]比a[j]大,取a[j]加入tmp, j++
{
tmp[k++] = a[j++];
ans += mid - i + 1; //a[i..mid]都比a[j]大,都能构成逆序对
}
else //取a[i]加入tmp, i++
tmp[k++] = a[i++];
}
//取完数组中剩余数字
while (i <= mid)
tmp[k++] = a[i++];
while (j <= r)
tmp[k++] = a[j++];
//复制回原数组
for (int i = l; i <= r; ++i)
a[i] = tmp[i];
}
void Merge_sort(int l, int r) //递归分治,归并排序
{
if (l < r)
{
int mid = (l+r) >> 1;
Merge_sort(l, mid);
Merge_sort(mid+1, r);
Merge(l, mid, r); //合并
}
}
int main()
{
int T, cnt;
cin >> T;
for(int kase = 1; kase <= T; ++kase)
{
cin >> cnt;
for (int i = 0; i < cnt; ++i)
cin >> a[i];
ans = 0;
Merge_sort(0, cnt-1);
printf("Scenario #%d:\n%d\n\n", kase, ans);
}
return 0;
}