将数组旋转k步

最新推荐文章于 2024-09-13 19:36:07 发布

blank_06

最新推荐文章于 2024-09-13 19:36:07 发布

阅读量47

点赞数

文章标签：数据结构算法前端 typescript

本文链接：https://blog.csdn.net/kd_want/article/details/129567359

版权

文章介绍了如何通过两种方法旋转数组，一种是使用pop和unshift，另一种是使用concat。在给定的K值（如4）下，对数组[1,2,3,4,5,6,7,8]进行旋转。第一种方法的时间复杂度较高，为O(n^2)，而第二种方法利用slice和concat，性能更优。文章强调了对数组操作性能的理解和优化的重要性。

摘要由CSDN通过智能技术生成

输入一个数组【1,2,3,4,5,6,7,8】

定义一个K值，假如为4

通过旋转K步输出结果数组为【5,6,7,8,1,2,3,4】

接下来提供两种思路进行实现

/**
 * 旋转数组k步 - 使用pop和unshift
 * @param arr arr
 * @param k k
 * @returns arr
 */

//思路一：将数组末尾元素挨个pop出来，再挨个unshift到前面去
function rotate1(arr:number[],k:number):number[]{
	const length = arr.length
	if(!k||length===0) return arr
	const step = Math.abs(k%length)//取绝对值
	for (let i=0;i<step;i++){
		const n = arr.pop()
		if (n!=null){
			arr.unshift(n)
		}
	}
	return arr
}

该思路时间复杂度为O（n^2）

其中遍历时间复杂度为O（n）

但是要注意的是内部的数组API的unshift时间复杂度也是O（n）

数组是有序连续的内存结构，其中shift、unshift、splice操作都是很慢的

/**
 * 旋转数组k步 - 使用concat
 * @param arr arr
 * @param k k
 * @returns arr
 */

//思路二：将数组后三个元素和前三个元素拿出来再拼接在一起
function rotate2(arr:number[],k:number):number[]{
	const length = arr.length
	if(!k||length===0) return arr
	const step = Math.abs(k%length)//取绝对值
	//slice不会动原数组 也不会很慢 O(1)
	const part1 = arr.slice(-step)
	const part2 = arr.slice(0,length-step)
	const part3 = part1.concat(part2)
	return part3
}