华为机考真题 -- 整数对最小和

题目描述：

给定两个整数数组array1、array2，数组元素按升序排列。假设从array1、array2中分别取出一个元素可构成一对元素，现在需要取出k对元素，并对取出的所有元素求和，计算和的最小值。

注意：
两对元素如果对应于array1、array2中的两个下标均相同，则视为同一对元素。

输入描述：

输入两行数组array1、array2，
每行首个数字为数组大小size(0 < size <= 100);0 < array1[i] <= 1000；0 < array2[i] <= 1000；
接下来一行为正整数k，0 < k <= array1.size() * array2.size()

输出描述：

满足要求的最小和

示例：

输入
3 1 1 2
3 1 2 3
2

输出
4
说明：用例中，需要取2对元素取第一个数组第0

C++源码：

#include <iostream>
#include <string>
#include <ctime>
#include <vector>
#include <algorithm>
#include <unordered_map>
#include <map>
#include <sstream>
#include <utility>
using namespace std;

int main() {
    int size1, size2, k;
    cin >> size1;
    vector<int> array1(size1);
    for (int i = 0; i < size1; ++i) {
        cin >> array1[i];
    }
    cin >> size2;
    vector<int> array2(size2);
    for (int i = 0; i < size2; ++i) {
        cin >> array2[i];
    }
    cin >> k;

    // 合并两个数组并排序
    vector<pair<int, bool>> merged;
    for (int num : array1) merged.push_back({ num, true });
    for (int num : array2) merged.push_back({ num, false });
    sort(merged.begin(), merged.end());

    int minSum = 0;
    int count = 0;
    for (const auto& pair : merged) {
        if (count == k) break;
        if (pair.second) { // 来自array1
            if (!array2.empty() && pair.first >= array2.front()) {
                minSum += pair.first + array2.front();
                array2.erase(array2.begin());
                ++count;
            }
        }
        else { // 来自array2，已确保array1中存在更小或相等的数（由排序保证）
            minSum += pair.first + array1.front();
            ++count;
        }
    }

    cout << minSum << endl;

    system("pause");

    return 0;
}