Python中将所有1分组的最小交换次数


假设我们有一个二进制数组数据,我们需要找到将数组中所有1集中到一起(数组中的任何位置)所需的最小交换次数。例如,如果数组是[1,0,1,0,1,0,0,1,1,0,1],则输出为3,因为可能的解决方案是[0,0,0,0,0,1,1,1,1,1,1]

为了解决这个问题,我们将遵循以下步骤:

  • one := 0, n := 数据数组的长度
  • 创建一个大小为n的数组summ,并将其填充为0,设置summ[0] := data[0]
  • one := one + data[0]
  • 对于i从1到n-1
    • summ[i] := summ[i - 1] + data[i]
    • one := one + data[i]
  • ans := one
  • left := 0, right := one – 1
  • 当right < n
    • 如果left为0,则temp := summ[right],否则temp := summ[right] – summ[left - 1]
    • ans := ans和one – temp中的最小值
    • right和left分别加1
  • 返回ans

示例(Python)

让我们来看下面的实现,以便更好地理解:

 在线演示

class Solution(object):
   def minSwaps(self, data):
      one = 0
      n = len(data)
      summ=[0 for i in range(n)]
      summ[0] = data[0]
      one += data[0]
      for i in range(1,n):
         summ[i] += summ[i-1]+data[i]
         one += data[i]
      ans = one
      left = 0
      right = one-1
      while right <n:
         if left == 0:
            temp = summ[right]
         else:
            temp = summ[right] - summ[left-1]
         ans = min(ans,one-temp)
         right+=1
         left+=1
      return ans
ob = Solution()
print(ob.minSwaps([1,0,1,0,1,0,0,1,1,0,1]))

输入

[1,0,1,0,1,0,0,1,1,0,1]

输出

3

更新于:2020年4月30日

261 次查看

开始您的职业生涯

完成课程获得认证

开始学习
广告