Python划分数组为连续数字集合的练习

下面我来为你详细讲解Python划分数组为连续数字集合的练习的完整攻略。

题目描述

给定一个升序的整数数组，将数组中连续的数字段组成一个新的二维数组，即如果元素之间差值为1，则当做一段，将所有相关的连续数字组合成一个列表。

例如，给定数组为[1,2,3,5,6,7,9,10]，则要求将该数组划分为[[1,2,3],[5,6,7],[9,10]]。

思路分析

该题的难度并不大，主要思路是遍历整个数组，将连续的数字段保存到结果列表中即可。

可以采用迭代法来解决：

1. 初始化一个结果列表 res，以及一个当前序列列表 curSeq
2. 从左到右遍历整个数组，将当前元素加入到当前序列 curSeq 中
3. 如果当前序列 curSeq 的最后两个元素之差不等于1，则表示当前序列已经结束，将当前序列 curSeq 加入到结果列表 res 中，并将当前序列设置为空列表
4. 遍历完成后，如果序列 curSeq 非空，则将其加入到结果列表 res 中
5. 返回结果列表 res

代码实现

下面是该算法的 Python 代码实现：

def partition_array(nums):
    res = []  # 保存结果的二维列表
    cur_seq = []  # 当前序列的列表
    for num in nums:
        if not cur_seq or num - 1 == cur_seq[-1]:  # 如果当前序列为空或者当前元素与当前序列最后一个元素差值等于1，则将其加入到当前序列中
            cur_seq.append(num)
        else:
            res.append(cur_seq)  # 否则将当前序列加入到结果列表中
            cur_seq = [num]  # 重新设置当前序列为只包含当前元素的列表
    if cur_seq:
        res.append(cur_seq)  # 处理最后一个序列
    return res

示例说明

下面分别对两个示例进行说明。

示例一

输入：

[1,2,3,5,6,7,9,10]

输出：

[[1, 2, 3], [5, 6, 7], [9, 10]]

解析：

上述代码遍历整个数组，第一个元素 1 为连续数字的起始点，将其加入到当前序列 curSeq 中，然后遍历到第二个元素 2，将其加入到当前序列中。遍历到第三个元素 3 后，发现当前序列 curSeq 中的最后一个元素与该元素之差为1，所以将当前元素加入到当前序列中。接下来遍历到第四个元素 5，则当前序列断开，将当前序列 curSeq 加入到结果列表 res 中，并重新设置当前序列为包含当前元素 5 的列表。以此类推，最终得到结果为 [[1,2,3],[5,6,7],[9,10]]。

示例二

输入：

[1, 3, 4, 5, 7, 8, 9, 10]

输出：

[[1], [3, 4, 5], [7, 8, 9, 10]]

解析：

同样地，上述代码遍历整个数组，第一个元素 1 为连续数字的起始点，将其加入到当前序列 curSeq 中，接着遍历到第二个元素 3，将其加入到当前序列中。遍历到第三个元素 4 后，发现当前序列 curSeq 中的最后一个元素与该元素之差为1，所以将当前元素加入到当前序列中。接下来遍历到第四个元素 5，则当前序列断开，将当前序列 curSeq 加入到结果列表 res 中，并重新设置当前序列为只包含当前元素 7 的列表。以此类推，最终得到结果为 [[1], [3,4,5],[7,8,9,10]]。