输入格式 第一行有三个数,分别是长度m、长度n和k,中间空格相连(1<=m,n<=100000; 1<=k<=m+n)。 第二行m个数分别是非减序的序列X。第三行n个数分别是非减序的序列Y。 输出格式 序列X和Y的第k小的数。 输入样例 5 6 7 1 8 12 12 21 4 12 20 22 26 31 输出样例 20
#include <iostream> using namespace std; void K(int ln, int lm, int k, int n[], int m[]) { //表示第 count 的数 int count = 0; int i = 0, j = 0; int result = 0; while (i < ln && j < lm) { if (n[i] < m[j]) { count ++; if (count == k) { result = n[i]; break; } i ++; } else { count ++; if (count == k) { result = m[j]; break; } j ++; } } while (i < ln) { count ++; if (count == k) { result = n[i]; break; } i ++; } while (j < lm) { count ++; if (count == k) { result = m[j]; break; } j ++; } cout << result << endl; } int main() { int ln, lm, k; int n[100000],m[100000]; cin >> ln >> lm >> k; for (int i = 0; i < ln; i ++) { cin >> n[i]; } for (int i = 0; i < lm; i ++) { cin >> m[i]; } K(ln, lm, k, n, m); return 0; }
测试用例:ios
5 6 7 1 8 12 12 21 4 12 20 22 26 31
输出结果测试
20