最少交换次数python_LeetCode - Python - 1151. 最少交换次数来组合所有的 1 - 期权论坛...

给出一个二进制数组 data，你需要通过交换位置，将数组中 任何位置 上的 1 组合到一起，并返回所有可能中所需 最少的交换次数。

示例 1：

输入：[1,0,1,0,1]

输出：1

解释：

有三种可能的方法可以把所有的 1 组合在一起：

[1,1,1,0,0]，交换 1 次；

[0,1,1,1,0]，交换 2 次；

[0,0,1,1,1]，交换 1 次。

所以最少的交换次数为 1。

示例 2：

输入：[0,0,0,1,0]

输出：0

解释：

由于数组中只有一个 1，所以不需要交换。

示例 3：

输入：[1,0,1,0,1,0,0,1,1,0,1]

输出：3

解释：

交换 3 次，一种可行的只用 3 次交换的解决方案是 [0,0,0,0,0,1,1,1,1,1,1]。

提示：

1 <= data.length <= 10^5

0 <= data[i] <= 1

第一种思路：

滑动窗口，

sum(data) 可以得到数组里 1 的个数，

然后开一个 sum(data) 长度的滑动窗口 window，

sum(window)就可以得到 window 里 1 的个数，

sum(data) - sum (window) 就是这个窗口需要的移动步数，

再从所有窗口的移动步数里找到最小的那个。

class Solution(object):

def minSwaps(self, data):

"""

:type data: List[int]

:rtype: int

"""

#滑动窗口

summ = sum(data)

res = summ

for i in range(len(data) - summ + 1):

window = data[i:i + summ]

# print window

res = min(res, summ - sum(window))

return res

第二种思路：

小小的优化一下，

每一个window不再用sum(window)计算window里 1 的个数，

而是通过局部的调整，减去第一个元素的值，然后加上最后一个元素的下一个元素的值，

来得到新 window 的 sum(window)。

class Solution(object):

def minSwaps(self, data):

"""

:type data: List[int]

:rtype: int

"""

#滑动窗口

summ = sum(data)

res = summ

window_sum = sum(data[:summ])

for i in range(len(data) - summ):

res = min(res, summ - window_sum)

window_sum = window_sum - data[i] + data[i + summ]

return res