7-数组-轮转数组

这是数组的第7题,力扣链接。

给定一个整数数组 nums,将数组中的元素向右轮转 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]

示例 2:

输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释: 
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]

在做链表的旋转时其实做过类似的,不过当时的最优解是把链表做成环,这里当然也可以,但就不这样做了,因为逻辑可能会比较复杂,需要维护额外的指针。

看到这道题我的第一反应是拿切片做,因为本质是两个切片互换顺序,由于需要更改原数组因此这里需要一个额外的赋值操作。

func rotate(nums []int, k int) {
	k %= len(nums)
	result := make([]int, 0)
	result = append(result, nums[len(nums)-k:]...)
	result = append(result, nums[:len(nums)-k]...)
	copy(nums, result)
}

还有一种思路是先将数组全反,然后旋转前k个和后k个。

func reverse(a []int) {
	for i, n := 0, len(a); i < n/2; i++ {
		a[i], a[n-1-i] = a[n-1-i], a[i]
	}
}

func rotate(nums []int, k int) {
	k %= len(nums)
	reverse(nums)
	reverse(nums[:k])
	reverse(nums[k:])
}

你可能感兴趣的:(白话算法,算法,数据结构)