题目描述
给定两个整数数组 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())
示例
示例1
输入
3 1 1 2
3 1 2 3
2
输出
4
解题思路
1、组合出所有可能的整数对,并求和
2、对所有求和结果升序排序
3、对最小的k个和进行求和
C++代码实现
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
vector<int> arr1, arr2;
int k;
char c;
while ((cin >> k).get(c)) {
arr1.emplace_back(k);
if(c == '\n')
break;
}
while ((cin >> k).get(c)) {
arr2.emplace_back(k);
if (c == '\n')
break;
}
cin >> k;
int i, j;
vector<int> res;
for (i = 0; i < arr1.size(); ++i) {
for (j = 0; j < arr2.size(); ++j) {
res.emplace_back(arr1[i] + arr2[j]);
}
}
sort(res.begin(), res.end());
j = 0;
for (i = 0; i < k; ++i) {
j += res[i];
}
cout << j << endl;
return 0;
}