title
给你两个整数数组 nums1 和 nums2 ,两个数组长度都是 n ,再给你一个整数 k 。你可以对数组 nums1 进行以下操作:
- 选择两个下标
i和j,将nums1[i]增加k,将nums1[j]减少k。换言之,nums1[i] = nums1[i] + k且nums1[j] = nums1[j] - k。
如果对于所有满足 0 <= i < n 都有 num1[i] == nums2[i] ,那么我们称 nums1 等于 nums2 。
请你返回使 nums1 等于 nums2 的 最少 操作数。如果没办法让它们相等,请你返回 -1 。
示例 1:
输入:nums1 = [4,3,1,4], nums2 = [1,3,7,1], k = 3
输出:2
解释:我们可以通过 2 个操作将 nums1 变成 nums2 。
第 1 个操作:i = 2 ,j = 0 。操作后得到 nums1 = [1,3,4,4] 。
第 2 个操作:i = 2 ,j = 3 。操作后得到 nums1 = [1,3,7,1] 。
无法用更少操作使两个数组相等。
示例 2:
输入:nums1 = [3,8,5,2], nums2 = [2,4,1,6], k = 1
输出:-1
解释:无法使两个数组相等。
提示:
n == nums1.length == nums2.length2 <= n <= 1050 <= nums1[i], nums2[j] <= 1090 <= k <= 105
solution
任选两个元素,一个+k,一个-k,问最少操作次数,使这个数组和另一个数组相同。累计操作次数即可,注意k为0的情况
class Solution:
def minOperations(self, nums1: List[int], nums2: List[int], k: int) -> int:
op1, op2 = 0, 0
d = []
for a, b in zip(nums1, nums2):
d.append(a - b)
if sum(d) != 0: return -1
for v in d:
if k == 0:
# k==0时特判,不能相除法了。
if v != 0: return -1
continue
if abs(v) % k != 0: return -1
if v > 0:
op1 += v // k
else:
op2 += (-v) // k
if op1 != op2:
return -1
return op1