nth_element

官方文档

第二个参数的nth是指下标，例如带入0表示寻找下标为0，第1小的数值

示例程序

#include <iostream>
#include <algorithm>

using namespace std;

int k; // 第k+1小的元素，下标为k
int a[5] = {6, 1, 5, 2, 7};

int main()
{
    cin >> k;

    for (int i = 0; i < 5; ++ i) cout << a[i] << ' '; cout << endl;

    nth_element(a, a + k, a + 5);

    for (int i = 0; i < 5; ++ i) cout << a[i] << ' '; cout << endl;

    printf("第%d小的数为%d，下标为%d
", k + 1, a[k], k);

    return 0;
}

输出结果