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
广告