每日一题——4行Python代码实现PAT乙级1008 数组元素循环右移问题（举一反三+思想解读+逐步优化）四千字好文

本文链接：https://blog.csdn.net/weixin_44915521/article/details/139824504

一个认为一切根源都是“自己不够强”的INTJ

个人主页：用哲学编程-CSDN博客
专栏：每日一题——举一反三
 Python编程学习
 Python内置函数

我的写法

N,M=map(int,input().split())
nums=list(map(int,input().split()))
output=nums[N-M%N:]+nums[:N-M%N]
print(*output)

这段代码是Python语言编写的，主要功能是对一个列表进行循环移位操作。下面是对这段代码的点评和分析：

代码功能

输入处理：首先，代码通过input().split()获取用户输入的两个整数N和M，并使用map(int, ...)将其转换为整数。
列表处理：接着，代码再次使用input().split()获取用户输入的一系列整数，并同样使用map(int, ...)转换为整数列表nums。
循环移位：代码通过切片操作nums[N-M%N:] + nums[:N-M%N]实现对列表nums的循环移位。这里M%N确保了移位的步长不会超过列表的长度N。
输出结果：最后，使用print(*output)将移位后的列表元素以空格分隔的形式输出。

时间复杂度分析

输入处理：map(int, input().split())的时间复杂度为O(N)，其中N是输入的元素数量。
列表处理：同样，map(int, input().split())的时间复杂度为O(N)。
循环移位：切片操作的时间复杂度为O(N)，因为需要复制列表的一部分。
输出：print(*output)的时间复杂度为O(N)，因为需要输出每个元素。

因此，总的时间复杂度为O(N)。

空间复杂度分析

输入处理：空间复杂度为O(N)，因为需要存储输入的整数。
列表处理：空间复杂度为O(N)，因为需要存储整数列表。
循环移位：切片操作的空间复杂度为O(N)，因为需要创建一个新的列表来存储移位后的结果。

总的空间复杂度为O(N)。

总结

这段代码简洁高效地实现了列表的循环移位功能，时间复杂度和空间复杂度均为O(N)，适合处理中等大小的数据集。然而，如果列表非常大，切片操作可能会导致额外的内存使用，这在某些资源受限的环境中可能需要考虑优化。

我要更强

为了优化时间复杂度和空间复杂度，我们可以考虑不使用切片操作，因为切片操作会创建一个新的列表，这在处理大列表时可能会导致不必要的内存消耗。以下是几种优化方法：

方法一：使用循环移位

这种方法通过直接在原列表上操作，避免了创建新列表，从而优化了空间复杂度。

def rotate_list(nums, M):
    N = len(nums)
    M %= N  # 确保M在有效范围内
    # 使用循环移位
    for _ in range(M):
        last = nums[-1]
        for i in range(N - 1, 0, -1):
            nums[i] = nums[i - 1]
        nums[0] = last
    return nums

# 主程序
N, M = map(int, input().split())
nums = list(map(int, input().split()))
rotated_nums = rotate_list(nums, M)
print(*rotated_nums)

时间复杂度: O(N * M)，因为每个元素可能需要移动M次。空间复杂度: O(1)，除了输入输出外，没有使用额外的空间。

方法二：使用Python的deque

Python的deque支持高效的插入和删除操作，可以用来优化循环移位。

from collections import deque

def rotate_list(nums, M):
    nums = deque(nums)
    nums.rotate(M)  # 使用deque的rotate方法
    return list(nums)

# 主程序
N, M = map(int, input().split())
nums = list(map(int, input().split()))
rotated_nums = rotate_list(nums, M)
print(*rotated_nums)

时间复杂度: O(N)，因为deque的rotate方法是O(N)的。空间复杂度: O(N)，因为创建了一个新的deque。

方法三：使用列表切片和拼接

虽然这种方法的空间复杂度较高，但代码简洁，易于理解。

def rotate_list(nums, M):
    N = len(nums)
    M %= N  # 确保M在有效范围内
    return nums[-M:] + nums[:-M]

# 主程序
N, M = map(int, input().split())
nums = list(map(int, input().split()))
rotated_nums = rotate_list(nums, M)
print(*rotated_nums)

时间复杂度: O(N)，因为切片和拼接操作都是O(N)的。空间复杂度: O(N)，因为创建了一个新的列表。