最小的和 （AcWing周赛）

给定两个长度为 n 的整数序列 a1,a2,…,an 和 b1,b2,…,bn。

现在要对序列 a进行恰好 k1 次操作，对序列 b 进行恰好 k2次操作。

每次操作具体流程为选取序列中的一个元素，并将其加 1或减 1。

要求所有操作进行完毕以后，∑(ai−bi)^2  尽可能小。

原题链接

将其处理成 abs(ai - bi) ,进行 (k1+k2) 次操作，每次对当前差值最大的数减一。

用优先队列，每次将 abs(ai-bi) 最大值放在队首进行处理。

#include <bits/stdc++.h>
using namespace std;

const int N=100010;
typedef long long ll;
int n,k1,k2;
int a[N],b[N];
priority_queue <ll> q;
ll ans;

int main(){
	cin>>n>>k1>>k2;
	
	for(int i=0;i<n;i++) cin>>a[i];
	for(int i=0;i<n;i++) 
	{
		cin>>b[i];
		q.push(abs(a[i]-b[i]));
	}
	
	int k = k1+k2;
	for(int i=0;i<k;i++)
	{
		ll temp = q.top();
		q.pop();
		q.push(abs(temp-1));
	}
	
	while(!q.empty())
	{
		ll t = q.top();
		q.pop();
		ans += t*t;
	}
	
	cout<<ans<<endl;
	
	return 0;
}