189. 轮转数组

2022-09-29  本文已影响0人  王侦

题目地址(189. 轮转数组)

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

题目描述

给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。


示例 1:

输入: nums = [1,2,3,4,5,6,7], k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右轮转 1 步: [7,1,2,3,4,5,6]
向右轮转 2 步: [6,7,1,2,3,4,5]
向右轮转 3 步: [5,6,7,1,2,3,4]

前置知识

公司

思路

关键点

代码

Go Code:


func rotate(nums []int, k int)  {
    if nums == nil || len(nums) == 1 {
        return
    }
    n := len(nums)
    k = k % n
    reverse(nums, 0, n - k - 1)
    reverse(nums, n - k, n - 1)
    reverse(nums, 0, n - 1)
}

func reverse(nums []int, head int, tail int) {
    for head < tail {
        nums[head], nums[tail] = nums[tail], nums[head]
        head++
        tail--
    }
}

复杂度分析

令 n 为数组长度。

上一篇 下一篇

猜你喜欢

热点阅读