题意:从两个数组list1,list2中各选一个数,记为a, b。最多可以选择n次,求选择的n个a*b和的最大值。
思路:贪心算法,对于两个数组中的正数部分,每次选择list1,list2中最大的a和b。对于负数部分,每次选择list1,lsit2中最小的a和b。若一个为正,一个为负,则不选择。
#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;
bool cmp(int a, int b)
{
return a > b;
}
vector<int> list1;
vector<int> list2;
int main()
{
int nc, np, k;
scanf("%d", &nc);
for(int i = 0; i < nc; i++)
{
scanf("%d", &k);
list1.push_back(k);
}
sort(list1.begin(), list1.end(), cmp);
scanf("%d", &np);
for(int i = 0; i < np; i++)
{
scanf("%d", &k);
list2.push_back(k);
}
sort(list2.begin(), list2.end(), cmp);
int j = 0, i = 0;
long long sum = 0;
while(i < nc && j < np)
{
if(list1[i] > 0 && list2[j] > 0)
{
sum += list1[i]*list2[j];
j++;
i++;
}
else
break;
}
int x = nc-1, y = np-1;
while(x >= i && y >= j)
{
if(list1[x] < 0 && list2[y] < 0)
{
sum += list1[x]*list2[y];
x--;
y--;
}
else
break;
}
printf("%lld\n", sum);
return 0;
}