贪心(优先队列)-Minimize the error - CodeForces - 960B

贪心(优先队列)-Minimize the error - CodeForces - 960B

题意：

$$\begin{gathered} 给定两个长度为n的数组a和b，对a操作k_1次，对b操作k_2次，每次操作可以选择数组中的一个数，使其增大或减小1。 \\ 求经过一系列操作后，{\sum }_{i=1}^n(a_i-b_i{)}^2的最小值。 \end{gathered}$$

Input：
2 0 0
1 2
2 3

Output：
2


Input：
2 1 0
1 2
2 2

Output：
0


Input：
2 5 7
3 4
14 4

Output：
1

数据范围：

$$\begin{gathered} n\in [1,1000]，k1+k2\in [0,1000]，a_i,b_i\in [-10^6,10^6]。 \\ Timelimit：1000ms，Memorylimit：262144kB。 \end{gathered}$$

题解：

$$\begin{gathered} 对表达式{\sum }_{i=1}^n{x}_i^2>=\frac{({\sum }_{i=1}^n{x}_i{)}^2}{n}，当且仅当x_1=x_2=...=x_n时取等， \\ 由均值不等式，可知，最小值在n个数相等时取到，要使表达式的值尽量小，n个项大小应当尽量接近。 \\ 对本题而言，考察d_i=|a_i-b_i|，每次操作可以使得d_i增大或减小1，总操作数为k=k_1+k_2。 \\ 因此，最优解应当从数组d的最大值开始，从大到小开始进行操作，每次减小最大的d_{max}，直到k次操作完。 \end{gathered}$$

需要注意的是：

$$\begin{gathered} ①、一开始这样想的，给数组d排序，接着从大到小，每次给最大的d尽量的减小到0，即： \\ 若剩余的操作次数k>=d_i，就把d_i置0，剩余k-d_i次操作；若k<d_i，就把d_i减小k。 \\ 后来发现其实数组d中的值应当是动态变化的，在最大的d_{max}逐渐减小的过程中，d_{max}可能会减小到次小项， \\ 原来的次小项会成为d_{max}。所以，要逐渐减小最大的d，需要用到优先队列来解决这个问题。 \\ ②、在计算d平方时，极端情况下会爆int，所以为了方便起见，d数组定义为ll类型。 \end{gathered}$$

具体落实：

$$\begin{gathered} ①、计算数组d，将每一个d[i]存储到优先队列中。 \\ ②、“一次一次”的操作，取最大的d_{max}出队，将其减小1，再将其入队，重复操作直到k=0。 \\ ③、将队列中的所有元素出队，计算最终的平方和。 \end{gathered}$$

---

代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>
#define inf 0x7fffffff
#define ll long long
using namespace std;
const int N=1010;
int n,k1,k2;
int a[N],b[N];
ll d[N];
priority_queue<int> Q;

int main()
{
    scanf("%d%d%d",&n,&k1,&k2);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) {scanf("%d",&b[i]);d[i]=abs(b[i]-a[i]);Q.push(d[i]);}

    int k=k1+k2;
    while(k>0)
    {
        int tmp=Q.top();
        Q.pop();
        Q.push(abs(tmp-1));
        k--;
    }

    ll ans=0;
    while(!Q.empty())
    {
        ll tmp=Q.top();
        ans+=(tmp*tmp);
        Q.pop();
    }

    printf("%lld\n",ans);

    return 0;
}