这题真是天坑,我照着书上写的算法直接交TLE,第四五个样例总是过不了,但样例又太大了几十万的数据点,不知道该怎么找问题。不是研究算法的,直接记住怎么写算了。
#include <iostream>
#include <cstdio>
using namespace std;
const int MAX = 100000;
int arr[MAX];
int partition(int a[], int l, int r) {
if (l < r + 1)swap(a[l], a[(l + r) >> 1]);
int t = a[l];
while (l < r) {//主要是 a[r]>t 和 a[l]<t 的比较中不要加入=,我也不知道啥原因
while (l < r && a[r] > t) r--;
if (l < r) a[l++] = a[r]; //还有,这个判断必须加,否则还是TLE
while (l < r && a[l] < t) l++;
if (l < r) a[r--] = a[l];
}
a[l] = t;
return l;
}
void qsort(int a[], int l, int r) {
if (l < r) {
int pos = partition(a, l, r);
qsort(a, l, pos - 1);
qsort(a, pos + 1, r);
}
}
int main() {
int n;
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%d", &arr[i]);
qsort(arr, 0, n - 1);
for (int i = 0; i < n - 1; i++)
printf("%d ", arr[i]);
printf("%d\n", arr[n - 1]);
return 0;
}
5.17补充:
while (b < e) {
while (b < e && arr[e] > t) e--;//b < e不能省略!!!如果去掉的话,在类似[5 4 3 2 1]这种case中会在下面那个循环中发生内存泄漏!!!
if (b < e) arr[b++] = arr[e];
while (b < e && arr[b] < t) b++;
if (b < e) arr[e--] = arr[b];
}