利用python求数组中连续k个相同值的最大的k

原创于 2017-07-14 15:17:29 发布 · 1.1k 阅读

2 ·

CC 4.0 BY-SA版权

零散的学习笔记专栏收录该内容

6 篇文章

订阅专栏

博客探讨了如何使用Python高效地找出数组中最大连续k个相同值的k，对比了传统两层循环的低效方法，并提供了一行代码的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

例如，对于一个数组 $a=(a_1,a_2,...,a_n)$ ，我们想知道使得 $a_i=a_{i+1}=...=a_{i+k}=x$ 的最大是多少，最笨的办法是我们需要做两层循环：

# 这里假设x=1
aaa = [1, 1, 0, -1, 1, 1, 1, -1, -1, -1, -1, 0, 0]
len_a = len(aaa)
max_group = 1
for i in range(len_a):
    first = aaa[i]
    if first != 1:
        continue
    for j in range(i+1, len_a):
        if aaa[j] != first:
            if max_group<j-i:
                max_group = j-i
            break
print(max_group)

但其实，python里可以一行代码实现上面的功能：

import itertools
aaa = [1, 1, 0, -1, 1, 1, 1, -1, -1, -1, -1, 0, 0]

max_group = max([len(list(j)) for i,j in itertools.groupby(aaa) if i==1])