给你两个长度可能不等的整数数组 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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
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 <= s2
int 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;
}
};
144 ms 111 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!