这道题很显然,直接二分即可,代码也很好理解~
思路:将a[i]从小到大排序(学校的录取线),然后循环遍历b[i],找到第一个大于等于a[i]的元素,如果没有就将指向最后一个,如果是第一个。那就直接加上两者之差即可fabs(a[1]-b[i]),否则比较fabs(b[i]-a[res])和fabs(b[i]-a[res-1]) 取其中的最小值即可
之前有两个点一直A不了,一定要注意。如果没有比b[i]大的元素a[i],那就将res指向a[i]的最后一个元素。
下面是AC代码~
#include <iostream>
#include <cmath>
#include <algorithm>
typedef long long int ll;
using namespace std;
int main()
{
int m,n;
cin>>m>>n;
int a[m+1]; //学校的预计分数线
int b[n+1]; //考生的估分
for(int i=1;i<=m;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
cin>>b[i];
}
ll sum=0;//记录所有的不满意度之和
sort(a+1,a+m+1);
for(int i=1;i<=n;i++) //查找第一个学校大于b[i]的分数线
{
int begin=1,end=m;
int res=end;//如果a[1]--a[m]中没有>=b[i]的元素
while(begin<=end)
{
int mid=(begin+end)/2;
if(a[mid]>=b[i])
{
res=mid;
end=mid-1;
}
else
{
begin=mid+1;
}
}
if(res==1)
{
sum+=fabs(a[res]-b[i]);
}
else
{
sum+=min(fabs(a[res]-b[i]),fabs(a[res-1]-b[i]));
}
}
cout<<sum<<endl;
return 0;
}
当然,还有更简单的做法。为STL正名~ AC代码
#include <iostream>
#include <cmath>
#include <algorithm>
typedef long long int ll;
using namespace std;
int main()
{
int m,n;
cin>>m>>n;
int a[m+1]; //学校的预计分数线
int b[n+1]; //考生的估分
for(int i=1;i<=m;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
cin>>b[i];
}
ll sum=0;//记录所有的不满意度之和
sort(a+1,a+m+1);
for(int i=1;i<=n;i++)
{
int res=lower_bound(a+1,a+m+1,b[i])-a;// 找到第一个大于等于b[i]的元素的位置
if(res==m+1)
{
sum+=fabs(a[m]-b[i]);
}
else if(res==1)
{
sum+=fabs(a[res]-b[i]);
}
else
{
sum+=min(fabs(a[res]-b[i]),fabs(a[res-1]-b[i]));
}
}
cout<<sum<<endl;
return 0;
}