189. Rotate Array

https://leetcode.com/problems/rotate-array/

Rotate Array - LeetCode

Given an array, rotate the array to the right by k steps, where k is non-negative.

 

Example 1:

Input: nums = [1,2,3,4,5,6,7], k = 3

Output: [5,6,7,1,2,3,4]

Explanation: rotate 1 steps to the right: [7,1,2,3,4,5,6] rotate 2 steps to the right: [6,7,1,2,3,4,5] rotate 3 steps to the right: [5,6,7,1,2,3,4]

 

Example 2:

Input: nums = [-1,-100,3,99], k = 2

Output: [3,99,-1,-100]

Explanation: rotate 1 steps to the right: [99,-1,-100,3] rotate 2 steps to the right: [3,99,-1,-100]

 

Constraints:

• 1 <= nums.length <= 105
• -231 <= nums[i] <= 231 - 1
• 0 <= k <= 105

 

Follow up:

• Try to come up with as many solutions as you can. There are at least three different ways to solve this problem.
• Could you do it in-place with O(1) extra space?

 

Solution

class Solution:
    def rotate(self, nums, k):
        k = k % len(nums)
        nums[:] = nums[-k:] + nums[:-k]

• k가 nums보다 길 수 있으므로, nums의 길이로 나눈 나머지로 바꿔준다.
• nums 의 뒷부분과 앞부분을 합쳐준다.
• 여기서 [:] 를 해야 본래 nums의 값을 바꿀 수 있다.
• [:]를 쓰지 않으면 함수 내에서 참조된 nums만 바뀐다.
• 아래의 코드와 그림으로 더 자세히 알아보자.

class Solution:
    def rotate(self, nums, k):
        k = k % len(nums)
        nums[:] = nums[-k:] + nums[:-k]

S = Solution()
a = [1,2,3,4,5]
S.rotate(a, 3)

• [:]를 붙이면, nums가 a가 가리키는 요소들에 접근하여 직접적으로 값을 바꾼다.
• 때문에, 함수를 나가도 본래 리스트인 a 의 값들이 바뀐다.
• [:]를 붙이지 않으면, nums에 새로운 리스트가 할당된다.
• 때문에, 함수를 나가면 본래 리스트인 a 에는 변화가 없다.