题目:
输入一个序列,交换任意两个相邻元素,不超过k次。在交换之后,问最少的逆序对有多少个
输入:
第一行是n和k,1 <= n <= 10^5,0 <= k <= 10^9;第二行包括n个整数{a1,a2,...,an},0 <= ai <= 10^9.
输出:
最少的逆序对数量。
输入样例:
3 1
2 2 1
输出样例:
1
注意:
- 当k=0时可以用暴力法,对每个数和它后面的所有数一个一个进行比较,时间复杂度是O(n^2),但由于本题n最大为10^5,用暴力法会TLE。
- 不超过k次是指可以少于k次,而不是一定要k次。
- 所有相邻数中,只有交换逆序对才会影响逆序对的数量。
- 分cnt<=k和cnt>k讨论。
题解:
//归并排序:求逆序对
#include <stdio.h>
#define N 100005
int n, k, q[N], tmp[N], cnt = 0;
int merge_sort(int q[], int l, int r)
{
if (l == r) return 0;
int mid = (l + r) / 2, i = l, j = mid + 1, t = 0;
merge_sort(q, l, mid);
merge_sort(q, mid + 1, r);
while (i <= mid && j <= r)
{
if (q[i] > q[j]) {
tmp[t++] = q[j++];
cnt += mid - i + 1;
}
else tmp[t++] = q[i++];
}
while (i <= mid) tmp[t++] = q[i++];
while (j <= r) tmp[t++] = q[j++];
for (i = l, j = 0; i <= r; i++, j++) q[i] = tmp[j];
if (cnt <= k) return 0;
else return (cnt - k);
}
int main(void)
{
scanf("%d %d", &n, &k);
for (int i = 0; i < n; i++)
scanf("%d", &q[i]);
printf("%d\n", merge_sort(q, 0, n - 1));
return 0;
}
错误:
在写代码的过程中,函数中的tmp数组的下标习惯性定义为k,导致在return时k的值被修改,从而导致结果错误,要注意定义变量时名称不能重复。