您的位置:首页 > 其它

求两个有序序列的第k小

2016-01-04 00:00 134 查看
摘要: 利用归并算法,求两个序列的第k小

输入格式
第一行有三个数,分别是长度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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  算法 归并