Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Xoay mảng trong Python


Cho mảng A, xoay phải k bước. Ví dụ: A=[5,7,3,6,8,1,5,4], k=3 → [1,5,4,5,7,3,6,8].

Giải pháp

Các bước:

  • n = len(nums), k %= n.
  • nums[:] = nums[n-k:] + nums[:n-k].

Ví dụ

Triển khai Python:

class Solution(object): 
    def rotate(self, nums, k): 
        n = len(nums) 
        k%=n 
        nums[:] = nums[n-k:]+nums[:n-k] 
nums = [5,7,3,6,8,1,5,4] 
ob1 = Solution() 
ob1.rotate(nums, 3) 
print(nums)

Đầu vào

nums = [5,7,3,6,8,1,5,4] k = 3

Đầu ra

[1,5,4,5,7,3,6,8]