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:])
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值