485. 最大连续 1 的个数#

题目描述#

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

示例 1：

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

提示：

1 <= nums.length <= 10^5
nums[i] 不是 0 就是 1

解题思路#

核心思想#

用分组循环把数组按连续 1 分段：找到一段 1 的起点 i，内层 j 向后扩展直到遇到 0，段长度 j - i + 1 更新答案，然后 i 跳到 j+1。

分组示意（nums = [1,1,0,1,1,1]）：

1
[1,1] | 0 | [1,1,1]
2
  2           3   → 最大 3

算法步骤#

初始化 ans = 0，i = 0。
当 i < n 时：
- 若 nums[i] == 0，i++ 跳过；
- 否则 j = i，内层 while 扩展连续 1；
- ans = max(ans, j - i + 1)，i = j + 1。
返回 ans。

逐步举例#

以 nums = [1,1,0,1,1,1] 为例：

段	区间	长度
11	[0,1]	2
111	[3,5]	3

最大为 3。

复杂度分析#

时间复杂度：O(N)，单次线性扫描。
空间复杂度：O(1)。

代码实现#

1
from typing import List
2

3

4
class Solution:
5
    def findMaxConsecutiveOnes(self, nums: List[int]) -> int:
6
        n = len(nums)
7
        ans = 0
8
        i = 0
9
        while i < n:
10
            if nums[i] == 0:
11
                i += 1
12
                continue
13
            j = i
14
            while j + 1 < n and nums[j + 1] == 1:
15
                j += 1
16
            ans = max(ans, j - i + 1)
17
            i = j + 1
18
        return ans

1
import sys
2
from pathlib import Path
3

4
sys.path.insert(0, str(Path(__file__).parent))
5
import unittest
6

7
from solution import Solution
8

9

10
class TestSolution(unittest.TestCase):
11
    def test_example(self):
12
        self.assertEqual(Solution().findMaxConsecutiveOnes([1, 1, 0, 1, 1, 1]), 3)
13

14
    def test_all_ones(self):
15
        self.assertEqual(Solution().findMaxConsecutiveOnes([1, 1, 1]), 3)
16

17
    def test_all_zeros(self):
18
        self.assertEqual(Solution().findMaxConsecutiveOnes([0, 0, 0]), 0)
19

20

21
if __name__ == '__main__':
22
    unittest.main()