189. 轮转数组
-
[题目链接](189. 轮转数组 - 力扣(LeetCode))
-
解题思路:直接举例子,
nums = [1,2,3,4,5,6,7],k=3,步骤:先把5,6,7安排好,即5,6,7,4,1,2,3这是怎么来的?其实可以想象成两部分,一部分是1,2,3,4,另一部分是5,6,7先分别交换,1和5,2和6,3和7。然后5,6,7就已经完成了。剩下的部分就是4,1,2,3怎么操作了。又分成两部分,一部分是4,另一部分是1,2,3,先安排少的,也就是先把4安排好,即3,1,2,4,接下来又分成两部分,一部分是3,另一部分是1,2,先安排少的,变成2,1,3,又分成两部分,2和1,此时长度相同,所以直接交换,变成1,2,就是最终的结果了。- 所以就是分成两部分
[l1...r1]以及[l2...r2],此时r1 + 1 == l2 - 先交换短的,然后短的那部分固定了,然后又可以分成两部分
- 具体步骤看代码
- 还有一个细节,k再大,其实都是
k %= n,n是数组长度。
- 所以就是分成两部分
-
代码
class Solution: # 将[l1..r1,l2 ..r2]变成[l2...r2,l1..r1] 前者r1 + 1 == l2 后者r2 + 1 ==l1 def process(self, nums, l1, r1, l2, r2): len1 = r1 - l1 + 1 len2 = r2 - l2 + 1 # 把小的换到大的那边去 if len1 == len2: i = l1 j = l2 while i <= r1: tmp = nums[i] nums[i] = nums[j] nums[j] = tmp i += 1 j += 1 elif len1 < len2: # 左边小 i = l1 j = l2 + len2 - len1 while i <= r1: tmp = nums[i] nums[i] = nums[j] nums[j] = tmp i += 1 j += 1 self.process(nums, l1, r1, l2, r2 - len1) else: # 右边小 i = l1 j = l2 while j <= r2: tmp = nums[i] nums[i] = nums[j] nums[j] = tmp j += 1 i += 1 self.process(nums, l1 + len2, r1, l2, r2) def rotate(self, nums: List[int], k: int) -> None: n = len(nums) k %= n if k == 0: return self.process(nums, 0, n - k - 1, n - k, n - 1)

浙公网安备 33010602011771号