leetcode -- Rotate Array -- 重点

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

注意k step 可以大于len(nums). 所以一开始要取余数。

有很多种方法。这里要参考
http://bookshadow.com/weblog/2015/02/24/leetcode-rotate-array/
http://www.geeksforgeeks.org/array-rotation/
因为要in-place 所以不能申请新list

思路1 reverse办法

自己的这种code比较好理解

class Solution(object):
    def rotate(self, nums, k):
        n = len(nums)
        k %= n

        self.reverse(nums, 0, n - k)#注意这里不包括n - k, 但是包括0
        self.reverse(nums, n - k, n)#注意这里包括n - k, 但是不包括n
        self.reverse(nums, 0, n)#注意这里包括0, 但是不包括n

    def reverse(self, nums, start, end):#这里跟list的截断一样,只包括start,不包括end
        i, j = start, end - 1
        while i < j:
            nums[i], nums[j] = nums[j], nums[i]
            i += 1
            j -= 1

还可以采用xor的办法进行元素交换, 例如交换a和b

a ^= b
b ^= a
a ^= b
class Solution(object):
    def rotate(self, nums, k):
        n = len(nums)
        k %= n

        self.reverse(nums, 0, n - k)#注意这里不包括n - k, 但是包括0
        self.reverse(nums, n - k, n)#注意这里包括n - k, 但是不包括n
        self.reverse(nums, 0, n)#注意这里包括0, 但是不包括n

    def reverse(self, nums, start, end):#这里跟list的截断一样,只包括start,不包括end
        i, j = start, end - 1
        while i < j:
            nums[i] ^= nums[j]
            nums[j] ^= nums[i]
            nums[i] ^= nums[j]
            i += 1
            j -= 1

ref中的code

class Solution:
    # @param nums, a list of integer
    # @param k, num of steps
    # @return nothing, please modify the nums list in-place.
    def rotate(self, nums, k):
        n = len(nums)
        k %= n
        self.reverse(nums, 0, n - k)
        self.reverse(nums, n - k, n)
        self.reverse(nums, 0, n)

    def reverse(self, nums, start, end):
        for x in range(start, (start + end) / 2):
            nums[x] ^= nums[start + end - x - 1]
            nums[start + end - x - 1] ^= nums[x]
            nums[x] ^= nums[start + end - x - 1]

思路2 不是in place, 供参考

my code :

class Solution(object):
    def rotate(self, nums, k):
        """ :type nums: List[int] :type k: int :rtype: void Do not return anything, modify nums in-place instead. """


        k = k % len(nums)
        if k > 0 :
                a = nums[-k:]
                a.extend(nums[:-k])
                nums[:] = a

你可能感兴趣的:(LeetCode)