待优化_求出2的下标组成的子列表

最新推荐文章于 2020-12-11 11:43:01 发布

趣享Eureka

最新推荐文章于 2020-12-11 11:43:01 发布

阅读量157

点赞数 1

分类专栏：算法

本文链接：https://blog.csdn.net/sinat_40701582/article/details/109693383

版权

算法专栏收录该内容

49 篇文章 0 订阅

订阅专栏

输入:
sequence = [1, 1, 2, 3, 2, 2, 2, 3, 3, 4, 2, 2, 2, 2]

输出:
[[2], [4, 5, 6], [10, 11, 12, 13]]

解法1

sequence = [1, 1, 2, 3, 2, 2, 2, 3, 3, 4, 2, 2, 2, 2]


def my_squence(sequence):
    index_2_list = [index for index, value in enumerate(sequence) if value == 2]

    queue = []
    result = []
    for i in index_2_list:
        if len(queue) == 0:
            queue.append(i)
        elif i == queue[-1] + 1:
            queue.append(i)
        else:
            result.append(queue[:])  # 测试发现queue调用clear()方法后,result中的queue 也被清除
            queue.clear()
            queue.append(i)  # 将下一个数(不等于上一个数+1)放入queue中
    if queue:
        result.append(queue)
    print(result)  


my_squence(sequence)