求两个有序序列的第k小

2019独角兽企业重金招聘Python工程师标准>>>

输入格式
第一行有三个数，分别是长度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

转载于:https://my.oschina.net/u/2301987/blog/596680