美文网首页
1151. 最少交换次数来组合所有的 1

1151. 最少交换次数来组合所有的 1

作者: 不湿的尿布湿 | 来源:发表于2020-03-02 00:35 被阅读0次

一、题目

给出一个二进制数组 data,你需要通过交换位置,将数组中 任何位置 上的 1 组合到一起,并返回所有可能中所需 最少的交换次数
难度:中等
链接:最少交换次数来组合所有的 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]。

三、解答

(一)错误:超出时间限制

class Solution:
    def minSwaps(self, data: List[int]) -> int:
        num1=data.count(1)
        NumWin=len(data)-num1+1
        num2 = [0]*(NumWin)
        answer = 0
        
        if num1 == 1:
            answer =0
        else:
            for i in range(NumWin):
                num2[i]=data[i:i+num1].count(1)
                answer = num1-max(num2)
            
        return answer
        

(二)正解:滑动窗口

class Solution:
    def minSwaps(self, data: List[int]) -> int:
        num1=data.count(1)
        NumWin=len(data)-num1+1
        num2 = [0]*(NumWin)
        answer = 0
        
        if num1 == 1:
            answer =0
        else:
            num2[0]=data[0:num1].count(1)
            for i in range(1,NumWin):
                num2[i]=num2[i-1]-data[i-1]+data[i+num1-1]
            answer = num1-max(num2)
            
        return answer

相关文章

网友评论

      本文标题:1151. 最少交换次数来组合所有的 1

      本文链接:https://www.haomeiwen.com/subject/kslkkhtx.html