试题 算法训练 最小乘积(基本型)
资源限制
时间限制:1.0s 内存限制:512.0MB
问题描述
给两组数,各n个。
请调整每组数的排列顺序,使得两组数据相同下标元素对应相乘,然后相加的和最小。要求程序输出这个最小值。
例如两组数分别为:1 3 -5和-2 4 1
那么对应乘积取和的最小值应为:
(-5) * 4 + 3 * (-2) + 1 * 1 = -25
输入格式
第一个行一个数T表示数据组数。后面每组数据,先读入一个n,接下来两行每行n个数,每个数的绝对值小于等于1000。
n<=8,T<=1000
输出格式
一个数表示答案。
样例输入
2
3
1 3 -5
-2 4 1
5
1 2 3 4 5
1 0 1 0 1
样例输出
-25
6
#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
int T,i,n,sum,a[10],b[10];
cin>>T;
while(T--){
cin>>n;
for(i=0;i<n;i++)
cin>>a[i];
for(i=0;i<n;i++)
cin>>b[i];
sort(a,a+n);
sort(b,b+n);
sum = 0;
for(i=0;i<n;i++)
sum+=a[i]*b[n-1-i];
cout<<sum<<endl;
}
return 0;
}
优化后:
#include <iostream>
#include <algorithm>
using namespace std;
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
int T,i,n,sum,a[10],b[10];
cin>>T;
while(T--){
cin>>n;
for(i=0;i<n;i++)
cin>>a[i];
for(i=0;i<n;i++)
cin>>b[i];
sort(a,a+n);
sort(b,b+n,cmp);
sum = 0;
for(i=0;i<n;i++)
sum+=a[i]*b[i];
cout<<sum<<endl;
}
return 0;
}
sort (a,a+n) 默认为从小到大排列
sort(a,a+n,cmp) 第三个参数是控制数组排列方式,是按从小到大还是从大到小排列
添加另一个函数
bool cmp(int a,int b)
{
return a>b;
} 则sort函数是按照从大到小的顺序排列的。