Swift - LeetCode - 最大连续 1 的个数

88 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第17天,点击查看活动详情

题目

给定一个二进制数组 nums, 计算其中最大连续 1 的个数。

示例 1:

  • 输入: nums = [1,1,0,1,1,1]
  • 输出: 3
  • 解释: 开头的两位和最后的三位都是连续 1 ,所以最大连续 1 的个数是 3.

示例 2:

  • 输入: nums = [1,0,1,1,0,1]
  • 输出: 2

提示:

  • nums[i] 不是 0 就是 1.

方法一:一次遍历

思路及解法

首先明确给定的数组是个二进制数组,也就是说数组中的元素只有 0 和 1,其次题目要求的是计算数组中 1 出现的连续次数的最大值,因此就明确了只要对元素 1 进行计数。为了得到数组中最大连续 1 的个数,需要遍历数组,并记录最大的连续 1 的个数和当前的连续 1 的个数。如果当前元素是 1,则将当前的连续 1 的个数加 1,否则,使用之前的连续 1 的个数更新最大的连续 1 的个数,并将当前的连续 1 的个数清零。

遍历数组结束之后,需要再次使用当前的连续 1 的个数更新最大的连续 1 的个数,因为数组的最后一个元素可能是 1,且最长连续 1 的子数组可能出现在数组的末尾,如果遍历数组结束之后不更新最大的连续 1 的个数,则会导致结果错误。

代码

class Solution {
    func findMaxConsecutiveOnes(_ nums: [Int]) -> Int {
        // count: 记录当前1连续出现的次数
        var count: Int = 0
        // max: 记录当前次数的最大值
        var maxCount: Int = 0
        var n: Int = nums.count
        for i in 0..<n {
            // nums[i] == 1, count++
            if nums[i] == 1 {
                count += 1
            }
            // 否则,count重新计数
            else {
                // 重置当前次数的最大值
                maxCount = max(maxCount, count)
                count = 0
            }
        }
        maxCount = max(maxCount, count)
        return maxCount
    }
}

复杂度分析

  • 时间复杂度:O(n)O(n),其中 nn 是数组的长度。需要遍历数组一次。

  • 空间复杂度:O(1)O(1)