求兩個有序序列的第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