在给定大小的组中反转数组 |第 2 组(第 1 组的变体)

题目描述

给定一个长度为n的数组和一个正整数k，将数组划分成大小为k的若干个组，并将每个组中的元素反转，最后返回修改后的数组。

示例

输入: nums = [1,2,3,4,5,6,7,8,9], k = 3
输出: [3,2,1,6,5,4,9,8,7]
解释:
划分后的每个组为 [1,2,3], [4,5,6], [7,8,9]，将每个组反转得到 [3,2,1], [6,5,4], [9,8,7]，最终得到的数组为 [3,2,1,6,5,4,9,8,7]。

解法

先将原数组划分成大小为k的组，对于每个组的元素进行反转，再拼接回原数组即可。

代码实现如下：

def reverse_groups(nums, k):
    # 划分成大小为k的组
    groups = [nums[i:i+k] for i in range(0, len(nums), k)]

    # 反转每个组中的元素
    for i in range(len(groups)):
        groups[i].reverse()

    # 拼接回原数组
    return [num for group in groups for num in group]

复杂度分析

• 时间复杂度：$O(n)$，其中n为数组的长度，遍历一遍数组进行划分和反转操作。
• 空间复杂度：$O(k)$，因为需要用一个大小为k的数组保存每个组的元素，再进行反转拼接操作。