LeetCode 1775. 通过最少操作次数使数组的和相等（贪心+双指针）

文章目录

- 1. 题目
- 2. 解题

1. 题目

给你两个长度可能不等的整数数组 nums1 和 nums2 。
两个数组中的所有值都在 1 到 6 之间（包含 1 和 6）。

每次操作中，你可以选择任意数组中的任意一个整数，将它变成 1 到 6 之间任意的值（包含 1 和 6）。

请你返回使 nums1 中所有数的和与 nums2 中所有数的和相等的最少操作次数。
如果无法使两个数组的和相等，请返回 -1 。

示例 1：
输入：nums1 = [1,2,3,4,5,6], nums2 = [1,1,2,2,2,2]
输出：3
解释：你可以通过 3 次操作使 nums1 中所有数的和与 nums2 中所有数的和相等。
以下数组下标都从 0 开始。
- 将 nums2[0] 变为 6 。 nums1 = [1,2,3,4,5,6], nums2 = [6,1,2,2,2,2] 。
- 将 nums1[5] 变为 1 。 nums1 = [1,2,3,4,5,1], nums2 = [6,1,2,2,2,2] 。
- 将 nums1[2] 变为 2 。 nums1 = [1,2,2,4,5,1], nums2 = [6,1,2,2,2,2] 。示例 2：
输入：nums1 = [1,1,1,1,1,1,1], nums2 = [6]
输出：-1
解释：没有办法减少 nums1 的和或者增加 nums2 的和使二者相等。示例 3：
输入：nums1 = [6,6], nums2 = [1]
输出：3
解释：你可以通过 3 次操作使 nums1 中所有数的和与 nums2 中所有数的和相等。
以下数组下标都从 0 开始。
- 将 nums1[0] 变为 2 。 nums1 = [2,6], nums2 = [1] 。
- 将 nums1[1] 变为 2 。 nums1 = [2,2], nums2 = [1] 。
- 将 nums2[0] 变为 4 。 nums1 = [2,2], nums2 = [4] 。提示：
1 <= nums1.length, nums2.length <= 10^5
1 <= nums1[i], nums2[i] <= 6

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/equal-sum-arrays-with-minimum-number-of-operations
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

排序，优先使用 sum 大的数组能降低的最多的，或者 sum 小的数组能升高最多的

class Solution {
public:int minOperations(vector<int>& nums1, vector<int>& nums2) {int len1 = nums1.size(), len2 = nums2.size();if(len1 > 6*len2 || len2 > 6*len1) return -1;sort(nums1.begin(), nums1.end());sort(nums2.begin(), nums2.end());int s1 = accumulate(nums1.begin(), nums1.end(),0);int s2 = accumulate(nums2.begin(), nums2.end(),0);if(s1 > s2){swap(nums1, nums2);swap(s1, s2);swap(len1, len2);}// 令 s1 <= s2int ans = 0, i = 0, j = len2-1;int delta1, delta2;while(s1 < s2 && (i < len1 || j >= 0)){delta1 = i < len1 ? 6-nums1[i] : 0;delta2 = j >= 0 ? nums2[j]-1 : 0;if(i < len1 && delta1 >= delta2){s1 += delta1;i++;ans++;}else if(j >= 0 && delta2 >= delta1){s2 -= delta2;j--;ans++;}}return ans;}
};