【周赛题解】6217.使数组相似的最少操作次数

回顾题目

给你两个正整数数组 nums 和 target ，两个数组长度相等。

在一次操作中，你可以选择两个 不同 的下标 i 和 j ，其中 0 <= i, j < nums.length ，并且：

令 nums[i] = nums[i] + 2 且
令 nums[j] = nums[j] - 2 。

如果两个数组中每个元素出现的频率相等，我们称两个数组是 相似 的。

请你返回将 nums 变得与 target 相似的最少操作次数。测试数据保证 nums 一定能变得与 target 相似。

示例 1：

输入：nums = [8,12,6], target = [2,14,10]
输出：2
解释：可以用两步操作将 nums 变得与 target 相似：
- 选择 i = 0 和 j = 2 ，nums = [10,12,4] 。
- 选择 i = 1 和 j = 2 ，nums = [10,14,2] 。
2 次操作是最少需要的操作次数。

示例 2：

输入：nums = [1,2,5], target = [4,1,3]
输出：1
解释：一步操作可以使 nums 变得与 target 相似：
- 选择 i = 1 和 j = 2 ，nums = [1,4,3] 。

示例 3：

输入：nums = [1,1,1,1,1], target = [1,1,1,1,1]
输出：0
解释：数组 nums 已经与 target 相似。

提示：

n == nums.length == target.length
1 <= n <= 105
1 <= nums[i], target[i] <= 106
nums 一定可以变得与 target 相似。

分析

题目保证一定有解，受到大佬灵茶山艾府的题解思路启发，在遇到这类题时先降低难度考虑题目的变种：如果将此题改为一次操作可以令 i 和 j 的元素各自 +1 和 -1该怎么做？

• 第一反应是将两个数组nums和target排序，然后从小到大依次求对应位置的两个数之差的绝对值。
• 因为一次操作包含了+1和-1，也就是将最后的绝对值之和除以2（1+1）就得出了总的操作次数。
• 因为排序过后小数对应小数，大数对应大数进行求差运算，因此得到的总的操作次数就是最小操作次数。

类比思想

了解了上面 +1/-1 的做法之后，我们再回过头来研究 +2/-2 的做法。

• 先将nums和target排序，因为只有排序过后，对应位置的两数之差的绝对值才能最小，最后总的操作次数才能达到最少。
• 每次操作一定会在原数上加2或者减2，因此不能直接在排序过后的1对1对应位置上进行操作，但是正好在一个数上加2或者减2不会改变这个数本身的奇偶性，因此我们可以将排序过后的数组按照奇偶性进行分组：奇数组内按照从小到大的顺序一一进行差值求和，偶数组内按照对应顺序一一进行差值求和，最后得到总的绝对值之和。
• 将总的绝对值之和除以4（2+2）就得到了总的操作次数，也是最少的操作次数，即题目所求答案。

题解代码+注释

class Solution {
public:
    long long makeSimilar(vector<int>& nums, vector<int>& target) {
        sort(nums.begin(),nums.end());
        sort(target.begin(),target.end());
        int js[2]={0};//受大佬启发的分组方式，用数组区分奇偶组的下标
        long long res=0;
        for(auto x:nums){
            int p=x%2;
            int &i=js[p];
            while(target[i]%2!=p)i++;//奇偶性不相同的直接跳过
            res+=abs(target[i]-x);//奇偶性相同的进行绝对值求差，res求和
            i++;
        }
        return res/4;
    }
};