leetcode316场周赛 第三题 使数组相等的最小开销 （两数组一块排序）

给你两个下标从 0 开始的数组 nums 和 cost ，分别包含 n 个 正 整数。

你可以执行下面操作 任意 次：

• 将 nums 中 任意 元素增加或者减小 1 。

对第 i 个元素执行一次操作的开销是 cost[i] 。

请你返回使 nums 中所有元素 相等 的 最少 总开销。

示例 1：

输入：nums = [1,3,5,2], cost = [2,3,1,14]
输出：8
解释：我们可以执行以下操作使所有元素变为 2 ：
- 增加第 0 个元素 1 次，开销为 2 。
- 减小第 1 个元素 1 次，开销为 3 。
- 减小第 2 个元素 3 次，开销为 1 + 1 + 1 = 3 。
总开销为 2 + 3 + 3 = 8 。
这是最小开销。

示例 2：

输入：nums = [2,2,2,2,2], cost = [4,2,8,1,3]
输出：0
解释：数组中所有元素已经全部相等，不需要执行额外的操作。

提示：

• n == nums.length == cost.length
• 1 <= n <= 10e5
• 1 <= nums[i], cost[i] <= 10e6

思路：

枚举nums数组的每一种可能，维护一个变化量var，就是从一个状态变化到下一个状态，每单位nums对应的权值变化是多少。

代码：

class Solution {
    typedef long long ll;
public:
    ll minCost(vector<int>& nums, vector<int>& cost) {
        int n = nums.size();
        vector<pair<int,int>> a(n);
        for(int i=0;i<n;i++)a[i]={nums[i],i};
        sort(a.begin(),a.end());
        ll ans = 1LL<<60,var = 0,tol = 0;
        //初始化，让所有nums变0
        for(int i=0;i<n;i++){
            var-=cost[a[i].second];
            tol+=(ll)a[i].first*cost[a[i].second];
        }
        for(int i=0;i<n;i++){
            var+=cost[a[i].second]*2;//到nums[i]以后，这个位置就该加了，所以反向增长一倍
            tol-=(ll)a[i].first*cost[a[i].second]*2;
            ans = min(ans,a[i].first*var+tol);
        }
        return ans;
    }
};