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. 1 <= data.length <= 10^5
2. 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