《算法笔记》上面的思路已经很简单了,但其实可以更无脑, 主要看每个人的敏感点在哪。我的代码如下:
#include <cstdio>
#include <algorithm>
using namespace std;
bool cmp(int a, int b){
if (a>=0 && b>=0) return a>b;
else return a<b;
}
int main(){
int nc,np;
int coupon[100010],value[100010];
scanf("%d",&nc);
for (int i=0; i<nc; i++){
scanf("%d",&coupon[i]);
}
scanf("%d",&np);
for (int i=0; i<np; i++){
scanf("%d",&value[i]);
}
sort(coupon, coupon+nc, cmp);
sort(value, value+np, cmp);
int sum=0;
int pos1=0, pos2=0;
while (pos1<nc && pos2<np){
if (coupon[pos1]*value[pos2]>=0){
sum+=coupon[pos1]*value[pos2];
pos1++;
pos2++;
}else if (coupon[pos1]>0 && value[pos2]<0){
pos2++;
}else {
pos1++;
}
}
printf("%d",sum);
}
和书上相比,这段代码感觉更容易理解一点点,更好写。