14. 整数对最小和

91 阅读1分钟

题目描述

给定两个整数数组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个元素与第二个数组第0个元素组成1对元素[1,1];取第一个数组第1个元素与第二个数组第0个元素组成1对元素[1,1];求和为1+1+1+1=4,为满足要求的最小和。

C++源码

暴力解题即可

#include <iostream>
#include <vector>
#include <sstream>
#include <algorithm>
#include <numeric>
using namespace std;

vector<int> split(string& s, char delimeter) {
    vector<int> res;
    string item;
    stringstream ss(s);
    while(getline(ss, item, delimeter)) {
        res.push_back(stoi(item));
    }
    return res;
}

int main()
{
    int n1;
    cin >> n1;

    vector<int>array1(n1);
    for (int i = 0; i < n1; i++) cin >> array1[i];

    int n2;
    cin >> n2;

    vector<int>array2(n2);
    for (int i = 0; i < n2; i++) cin >> array2[i];
    int k;
    cin >> k;
    vector<int> pairs;

    for (const auto &v1: array1) {
        for (const auto &v2: array2) {
            pairs.emplace_back(v1 + v2);
        }
    }

    sort(pairs.begin(), pairs.end());

    cout << accumulate(pairs.begin(), pairs.begin() + k, 0) << endl;
    return 0;
}