PAT甲级1037 Magic Coupon (25 分)

Posted 2020-11-13 ldudxy

题意：

输入一个正整数N（<=1e5），接下来输入N个整数。再输入一个正整数M（<=1e5），接下来输入M个整数。每次可以从两组数中各取一个，求最大的两个数的乘积的和。

代码：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
long long a[100007],b[100007];
long long c[100007],d[100007];
int cnta,cntb,cntc,cntd;
int main(){
int n,m;
cin>>n;
long long x;
for(int i=1;i<=n;++i){
cin>>x;
if(x>0)
a[++cnta]=x;
else if(x<0)
b[++cntb]=x;
}
cin>>m;
for(int i=1;i<=m;++i){
cin>>x;
if(x>0)
c[++cntc]=x;
else if(x<0)
d[++cntd]=x;
}
long long ans=0;
sort(a+1,a+1+cnta);
sort(b+1,b+1+cntb);
sort(c+1,c+1+cntc);
sort(d+1,d+1+cntd);
for(int i=1;i<=min(cntb,cntd);++i)
ans+=b[i]*d[i];
for(int i=cnta,j=cntc;i&&j;--i,--j)
ans+=a[i]*c[j];
cout<<ans;
return 0;
}