CodeForces 960 B.Minimize the error（思维）

题意

给你两个数字序列a，b，每个序列长度都为n，然后E=∑（a[i]-b[i]）^2。现在你可以改变a，b序列中的元素k1次和k2次，每次可以使一个元素加一或者减一。使得改变结束之后E的值最小。

题解

用一个数组来存入abs(a[i]-b[i])的值，有k1+k2的操作（+1/-1），以绝对值从小到大排，大于的-1，小于0的+1，暴力解决，不会超时，然后在遍历平方加起来。

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

const int maxn = 1005;
long long number1[maxn], number2[maxn], ans[maxn];

bool cmp(long long a, long long b){
	return abs(a) < abs(b);
}

int main(){
	int n, k1, k2;
	while(cin>>n>>k1>>k2){
		memset(ans, 0, sizeof(ans));
		memset(number2, 0, sizeof(number2));
		memset(number1, 0, sizeof(number1));
		for(int i = 0; i < n; i++){
			cin>>number1[i];
		}
		for(int i = 0; i < n; i++){
			cin>>number2[i]; 
		}
		for(int i = 0; i < n; i++){
			ans[i] = abs(number1[i]-number2[i]);
		}
		sort(ans, ans+n, cmp);
		int t = k1 + k2;
		while(t){
			if(ans[n-1] < 0){
				ans[n-1]++;
			}
			else{
				ans[n-1]--;
			}
			t--;
			sort(ans, ans+n, cmp);
		}
		long long sum = 0;
		for(int i = 0; i < n; i++){
			sum += ans[i]*ans[i];
		}
		cout<<sum<<endl;
	}
}