在 Python 中以最少的操作次数查找相等和数组的程序

假设我们有两个名为 nums1 和 nums2 的数组。数组中的值介于 1 到 6（含）之间。在一次操作中，我们可以将任何数组中的任何值更新为 1 到 6 之间的任何值。我们必须找到使 nums1 中的值之和等于 nums2 中值之和所需的最少操作次数。如果不可能，我们必须返回-1。

因此，如果输入类似于 nums1 = [1,5,6], nums2 = [4,1,1]，那么输出将为 2，因为我们可以将 nums2 从 [4,1,1] 更改为 [4， 1,6] 在第一次操作中，[4,2,6] 在第二次操作中使其等于 nums1。

示例

让我们看看以下实现以获得更好的理解 -

def solve(nums1, nums2):
   s1 = sum(nums1)
   s2 = sum(nums2)
   nums1.sort()
   nums2.sort()
   if s1>s2:
      nums1, nums2 = nums2, nums1
      s1, s2 = s2, s1

   ans = 0
   left, right = 0, len(nums2)-1
   while(left<len(nums1) or right>=0):
      if s1==s2:
         return ans
      curr_left = nums1[left] if left<len(nums1) else 7
      curr_right = nums2[right] if right>=0 else 0
      if 6-curr_left>=curr_right-1:
         s1+= min(6-curr_left, s2-s1)
         left+=1
      else:
         s2-= min(curr_right-1, s2-s1)
         right-=1
      ans+=1
   return -1 if s1!=s2 else ans

nums1 = [1,5,6]
nums2 = [4,1,1]
print(solve(nums1, nums2))

输入

[1,5,6], [4,1,1]

2