nth_element 使用方法

找1-n数组的第k小(k从1开始编号)

nth_element(a + 1, a + k, a + n + 1);
printf("%d\n", a[k]);

注意即使是1-n的数组,找第k小也是a+k而非a+k+1
如果是0-n-1的数组,找第k小为(k从0开始编号)

nth_element(a,a + k, a + n);
printf("%d\n", a[k];

复杂度O(n),执行完毕后a[k]就是第k小的元素