Python之旋转数组

旋转数组
给定一个数组,将数组中的元素向右移动K个位置,其中k是非负数
示例:
示例1
输入:numbs = [1,2,3,4,5,6,7,] k = 3
输出:[5,6,7,1,2,3,4]
过程:
第一步:[7,1,2,3,4,5,6]
第二步:[6,7,1,2,3,4,5]
第三步:[5,6,7,1,2,3,4] 输出
from typing import List


def rotate(nums: List[int], k: int) -> list[int]:
    """
    -> None:表示不返回任何值
    -> list[int] :表示返回列表
    """
    n = 0  # 旋转次数
    while n < k:
        c = nums.pop()  # 删除列表最后一位元素并返回赋值给变量c
        nums.insert(0, c)  # 在列表的0索引位置插入c变量
        n += 1  # 旋转次数更新
    return nums  # 将更新后的列表返回


ns = [1, 5, 0, 4, 6, 2, 11, 20, 35, -50]
k_value = 2
print('ns:', ns)
print('---')
print(f'旋转{k_value}个位置后:')
print(rotate(ns, k_value))
print('ns:', ns)
print('原始列表已经被更新')

你可能感兴趣的:(算法,数据结构,python)