问题链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805451374313472
题意:
给出两个集合,从这两个集合中分别选取相同数量的元素进行一对一相乘,问能得到的乘积之和最大是多少。
法一:两集合从小到大排序,访问负数从0开始,访问正数从末尾开始。即满足负数从小到达,正数从大到小的原则。
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 100010;
int coupon[maxn], product[maxn];
int main(){
int n, m;
cin >> n;
for(int i = 0; i < n; i++)
cin >> coupon[i];
cin >> m;
for(int i = 0; i < m; i++)
cin >> product[i];
sort(coupon, coupon + n);
sort(product, product + m);
int i = 0, j, ans = 0;
while(i < n && i < m && coupon[i] < 0 && product[i] < 0)
{
ans += coupon[i] * product[i];
i++;
}
//指向数组最后,即最大正数
i = n - 1;
j = m - 1;
while(i >= 0 && j >= 0 && coupon[i] > 0 && product[j] > 0)
{
ans += coupon[i] * product[j];
i--;
j--;
}
cout << ans << endl;
return 0;
}
法二:对两个个集合,将正负数分开考虑(0不做考虑)
然后对每个集合的正数按从大到小排序;负数按从小到大排序;
用vector容器,第一个集合的正数存放在pos[1]中,负数存放在neg[1]中。
第二个集合的正数存放在pos[2]中,负数存放在neg[2]中。
#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> pos[3];
vector<int> neg[3];
bool cmp(int a, int b){
return a > b;
}
int main(){
int n, m, x;
cin >> n;
for(int i = 0; i < n; i++){
cin >> x;
if(x > 0)
pos[1].push_back(x);
if(x < 0)
neg[1].push_back(x);
}
cin >> m;
for(int i = 0; i < m; i++){
cin >> x;
if(x > 0)
pos[2].push_back(x);
if(x < 0)
neg[2].push_back(x);
}
sort(pos[1].begin(), pos[1].end(), cmp);
sort(pos[2].begin(), pos[2].end(), cmp);
sort(neg[1].begin(), neg[1].end());
sort(neg[2].begin(), neg[2].end());
int ans = 0;
int i1 = 0, j1 = 0;
while(i1 < pos[1].size() && j1 < pos[2].size()){
ans += pos[1][i1] * pos[2][j1];
i1++; j1++;
}
int i2 = 0, j2 = 0;
while(i2 < neg[1].size() && j2 < neg[2].size()){
ans += neg[1][i2] * neg[2][j2];
i2++; j2++;
}
cout << ans << endl;
return 0;
}