2021.8.10提高B组模拟2T3 比赛（二分）（贪心—前缀和）

Posted 2022-01-10 SSL_LKJ

比赛

题目大意

输入样例

2
3 7
1 5

输出样例

20.0

题目数据

解题思路

首先答案肯定时两两匹配，结果再除以n

接下来就就是优化

我们可以先将 B 数组进行排序

对于每一个$a_i$有一些操作

在排好序的 B 数组中找到一个位置 k，使左边的数字小于它，右边的数字大于它（二分）

然后左边算（a[] - b[]）的平方，右边算（b[] - a[]）的平方

显然如果暴力做，还是会TLE

那我们就可以用 完全平方公式

那左边就成了

$$k\ast a_i\ast a_i-a_i\ast （b_1+b_2+\dots \dots +b_k)+（b_1\ast b_1+b_2\ast b_2+\dots \dots +b_k\ast b_k）$$

$b_1$~$b_k$的和 与 $b_1\ast b_1$~ $b_k\ast b_k$的和 可以用前缀和来预处理

就完成优化了

注：记得开 long long 和 特判

AC代码

#include<algorithm>
#include<cstdio>
using namespace std;
long long n,a[50005],b[50005],f[50005],fa[50005];
double ans;
void work(long long x)//二分

	long long l=1,r=n,k=0;
	if(a[x]>=b[n])//特判
	
		ans+=n*a[x]*a[x]-2ll*a[x]*f[n]+fa[n];
		return;
	
	if(a[x]<=b[1])
	
		ans-=n*a[x]*a[x]-2ll*a[x]*f[n]+fa[n];
		return;
		
	while(l<=r)
	
		long long mid=(l+r)/2ll;
		if(a[x]>=b[mid])l=mid+1,k=mid;
		else r=mid-1;
	
	ans+=k*a[x]*a[x]-2ll*a[x]*f[k]+fa[k]-((n-k)*a[x]*a[x]-2ll*a[x]*(f[n]-f[k])+(fa[n]-fa[k]));

int main()

	scanf("%lld",&n);
	for(long long i=1;i<=n;i++)scanf("%lld",&a[i]);
	for(long long i=1;i<=n;i++)scanf("%lld",&b[i]); 
	sort(b+1,b+n+1);
	for(long long i=1;i<=n;i++)//预处理
	
		f[i]=f[i-1]+b[i];
		fa[i]=fa[i-1]+b[i]*b[i];
	
	for(long long i=1;i<=n;i++)work(i);
	printf("%.1lf",ans/(n*1.0));
	return 0;

谢谢