codeforces 960B Minimize the error

题目链接：http://codeforces.com/contest/960/problem/B

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

分析：题目要求我们分别改变a，b序列的元素，但是我们需要求得是差值的平方和，因此我们可以转化为改变差值，那么就是一共在n个差值中改变k1+k2次，然后求E的最小值。我们只需要用优先队列存储两个序列对应位置的差值，然后每次改变队头的元素就可以解决这个问题。

AC代码：

#include<bits/stdc++.h>

using namespace std;

long long a[1005],b[1005];
priority_queue<int>q;
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n,k1,k2;
    while(cin>>n>>k1>>k2){
        for(int i=1;i<=n;i++){
            cin>>a[i];
        }
        for(int i=1;i<=n;i++){
            cin>>b[i];
        }
        for(int i=1;i<=n;i++){
            q.push(abs(a[i]-b[i]));
        }
        for(int i=1;i<=k1+k2;i++){
            long long p=q.top();
            q.pop();
            if(p==0){
                p++;
            }
            else p--;
            q.push(p);
        }
        long long result=0;
        while(!q.empty()){
            long long p=q.top();
            result+=p*p;
            q.pop();
        }
        cout<<result<<endl;
    }

return 0;
}