在C ++中从连接M次的数组中找到第K个最小元素

考虑我们有一个数组A，还有另外两个整数K和M。将数组与自身连接M次后，必须找到第K个最小元素。假设数组像A = [3，1，2]，K = 4且M = 3，所以在将A串联3次之后，它将是[3，1，2，3，1，2，3，1 ，2]，则第4个最小元素为2。

为解决此问题，我们将对数组A进行排序，然后返回数组索引（（K – 1）/ M）处的值。

示例

#include<iostream>
#include<algorithm>
using namespace std;
int findKSmallestNumber(int A[], int N, int M, int K) {
   sort(A, A + N);
   return (A[((K - 1) / M)]);
}
int main() {
   int A[] = { 3, 1, 2 };
   int M = 3, K = 4;
   int N = sizeof(A) / sizeof(A[0]);
   cout << K << "th smallest number after concatenating " << M << " times, is: "<<findKSmallestNumber(A, N, M, K);
}

输出结果

4th smallest number after concatenating 3 times, is: 2