PAT-A 1037. Magic Coupon (25)

题目链接在此。

题意

给定两个集合，从两个集合中分别选取相同数量的元素进行一对一相乘，求得到的最大乘积之和。

思路

要得到最大乘积和，采取如下贪心策略：
对于每一个集合，正数和负数分开考虑（0不用考虑，因为0乘任何数都为0，对和没有意义），两个集合正数和负数处于相同大小地位的数相乘相加，最后的结果就是最大和。

所以首先对输入的两个集合进行从大到小排序（从小到大也可以，只需要注意后面操作的对应顺序即可），这样一来，正数在前，负数在后，让两个集合从下标0开始对正数部分相乘相加，直到某一个集合出现负数或0；然后从两个集合的最后一个元素对应位置元素相乘相加， 知道某一个集合出现正数或0结束。最后的累加和就是结果。

AC代码

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

long long c[100010], p[100010];

//从大到小排列 
bool cmp(long long a, long long b){
    return a>b;
}

int main(){

    int NC,NP;

    scanf("%d",&NC);    
    for(int i = 0; i < NC; i++){
        scanf("%lld",&c[i]);
    }

    scanf("%d",&NP); 
    for(int i = 0;i < NP; i++){
        scanf("%lld",&p[i]);
    }

    sort(c,c+NC,cmp);
    sort(p,p+NP,cmp);

    int index_c = 0, index_p = 0;
    long long sum = 0;
    while(c[index_c]>0 && p[index_p]>0){
        sum += c[index_c]*p[index_p];
        index_c++;
        index_p++; 
    }

    index_c = NC-1; 
    index_p = NP-1;
    while(c[index_c]<0 && p[index_p]<0){
        sum += c[index_c]*p[index_p];
        index_c--;
        index_p--; 
    }

    printf("%lld\n",sum);

    return 0;
}