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 ^= bclass 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 -= 1ref中的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