[LeetCode] 189. Rotate Array

문제링크

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

Rotate Array - LeetCode

 

코드 

풀이1

class Solution:
    def rotate(self, nums: List[int], k: int) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        k = k % n
        nums[:] = nums[n-k:] + nums[:n-k]

 

풀이2

class Solution:
 
    def rotate(self, nums: List[int], k: int) -> None:
        def reverse(nums, i, j):
            while i<j:
                nums[i],nums[j] = nums[j], nums[i]
                i+=1
                j-=1
    
        """
        Do not return anything, modify nums in-place instead.
        """
        n = len(nums)
        k = k % n
        reverse(nums, 0, n-1)
        reverse(nums, 0, k-1)
        reverse(nums, k, n-1)

 

POINT

풀이1:

리스트 슬라이싱을 사용해서 필요한 부분만큼 잘라 리스트 앞에 붙였다.

 

풀이2:

reverse를 사용한 방법이다.

총 3번 reverse를 하는데,

첫번째는 0~끝까지 뒤집고, 두번쨰는 0~k-1까지 , 세번째는 k~끝까지 뒤집는다.