在 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=0):
if s1==s2:
return ans
curr_left = nums1[left] if left=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