输入格式
第一行有三个数,分别是长度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;
}
测试用例:
5 6 7
1 8 12 12 21
4 12 20 22 26 31
输出结果
20