#include<bits/stdc++.h>
using namespace std;
vector <int> v;
int a[105000];
int partition(vector<int>& arr, int low, int high) {
int pivot = arr[low];
while (low < high) {
while(low < high && arr[high] >= pivot) --high;
arr[low] = arr[high];
while(low < high && arr[low] < pivot) ++low;
arr[high] = arr[low];
}
arr[low] = pivot;
return low;
}
vector<int> getLeastNumbers(vector<int>& arr, int k) {
vector<int> vec(k, 0);
if (!k) return vec; // 排除 0 的情况
int n = (int)arr.size(), pos = partition(arr, 0, n - 1);
int low = 0, high = n - 1;
while (pos != k - 1) {
if (pos > k - 1) {
high = pos - 1;
pos = partition(arr, low, high);
}
if (pos < k - 1) {
low = pos + 1;
pos = partition(arr, low, high);
}
}
for (int i = 0; i < k; ++i) vec[i] = arr[i];
return vec;
}
int main(){
int n,k,i;
scanf("%d %d",&n,&k);
for (i=1;i<=n;i++) scanf("%d",&a[i]);
for (i=1;i<=n;i++) v.push_back(a[i]);
vector<int>ans=getLeastNumbers(v,k);
for (i=0;i<ans.size();i++) printf("%d ",ans[i]);
}
var
n,i:longint;
a:array[0..2000000]of qword;
procedure kp(l,r:longint);
var
i,j,mid:longint;
begin
if l>=r then exit;
i:=l;j:=r;mid:=a[(l+r) div 2];
repeat
while a[i]<mid do inc(i);
while a[j]>mid do dec(j);
if i<=j then
begin
a[0]:=a[i];a[i]:=a[j];a[j]:=a[0];
inc(i);dec(j);
end;
until i>j;
kp(l,j);
kp(i,r);
end;
begin
readln(n);
for i:=1 to n do
read(a[i]);
kp(1,n);
for i:=1 to n do
writeln(a[i]);
end.
原文:https://www.cnblogs.com/nowheretrix/p/12463167.html