【数据结构-距离和】力扣2602. 使数组元素全部相等的最少操作次数

给你一个正整数数组 nums 。

同时给你一个长度为 m 的整数数组 queries 。第 i 个查询中，你需要将 nums 中所有元素变成 queries[i] 。你可以执行以下操作 任意 次：

将数组里一个元素 增大 或者 减小 1 。
请你返回一个长度为 m 的数组 answer ，其中 answer[i]是将 nums 中所有元素变成 queries[i] 的 最少 操作次数。

注意，每次查询后，数组变回最开始的值。

示例 1：
输入：nums = [3,1,6,8], queries = [1,5]
输出：[14,10]
解释：第一个查询，我们可以执行以下操作：

• 将 nums[0] 减小 2 次，nums = [1,1,6,8] 。
• 将 nums[2] 减小 5 次，nums = [1,1,1,8] 。
• 将 nums[3] 减小 7 次，nums = [1,1,1,1] 。
  第一个查询的总操作次数为 2 + 5 + 7 = 14 。
  第二个查询，我们可以执行以下操作：
• 将 nums[0] 增大 2 次，nums = [5,1,6,8] 。
• 将 nums[1] 增大 4 次，nums = [5,5,6,8] 。
• 将 nums[2] 减小 1 次，nums = [5,5,5,8] 。
• 将 nums[3] 减小 3 次，nums = [5,5,5,5] 。
  第二个查询的总操作次数为 2 + 4 + 1 + 3 = 10 。

示例 2：
输入：nums = [2,9,6,3], queries = [10]
输出：[20]
解释：我们可以将数组中所有元素都增大到 10 ，总操作次数为 8 + 1 + 4 + 7 = 20 。

以上图引用自：

class Solution {
public:
    vector<long long> minOperations(vector<int>& nums, vector<int>& queries) {
        ranges::sort(nums);
        int n =nums.size();
        vector<long long> sum(n + 1);
        for(int i = 0; i < n;i++){
            sum[i + 1] = sum[i] + nums[i];
        }

        int m = queries.size();
        vector<long long> ans(m);
        for(int j = 0;j < m; j++){
            int q = queries[j];
            long long k = ranges::lower_bound(nums, q) - nums.begin();
            long long left = q * k - sum[k];
            long long right = sum[n] - sum[k] - (n-k) * q;
            ans[j] = right + left;
        }
        return ans;
    }
};

时间复杂度：O((n+q)logn)，其中 n 为 nums 的长度，q 为 queries 的长度。
空间复杂度：O(n)。返回值不计入。

这道题目首先计算前缀和，然后对queries的元素为参数一一使用二分查找。

 long long k = ranges::lower_bound(nums, q) - nums.begin();

减去nums.begin()，也就是两个迭代器相减，可以算出他们之间的距离，这时候k也就是比q大的第一个元素的索引值。

然后计算左边部分的蓝色面积和右边部分的绿色面积，最后相加后得到的答案就是需要操作的次数。