将一个数组旋转k步

 

1、注意算法时间复杂度

2、识破内部API的时间复杂度，如unshift

3、单元测试，考虑参数非法情况，提升代码健壮性

重点

1、优先考虑时间复杂度

2、比复杂度更重要的是：代码逻辑清晰，易读

 

 

1、数组方法：pop，unshift

export function rotate1(arr: number[], k:number): number[] {
    const len = arr.length
    if (!k || len === 0) {
        return arr
    }
    const step = Math.abs(k % len)
    for (let i = 0; i< step; i++) { // o(n)
      const n = arr.pop();
       if (n !== null) {
        // 数组是一个有序结构（连续的内存空间），unshift、shift、splice会非常慢（截取重新排序），push很快（后面加一个）
           arr.unshift(n) // o(n)
       }
    }
    return arr
} // o(n^2)
 
// 功能测试
// const arr = [1,2,3,4,5,6,7]
// const arrRotate = rotate1(arr, 3)
// console.log(arrRotate)

 

2、拆数组a1、a2：a2.concat(a1)

export function rotate2(arr: number[], k:number): number[] {
    const len = arr.length
    if (!k || len === 0) {
        return arr
    }
    const step = Math.abs(k % len)

    const part1 = arr.slice(-step) // 原数组不变 o(1)
    const part2 = arr.slice(0, len-step)
    const result = part1.concat(part2)

    return result
}

// const arrRotate2 = rotate2(arr, 3)
// console.log(arrRotate2)

 

3、性能测试

const arr = []
for (let i = 0; i < 10 * 10000; i++) {
    arr.push(i)
}

console.time('rotate1')
rotate1(arr, 9 * 10000) // 885ms 时间复杂度o(n^2)
console.time('rotate1')

const arr1 = []
for (let i = 0; i < 10 * 10000; i++) {
    arr1.push(i)
}

console.time('rotat2')
rotate2(arr1, 9 * 10000) // 1ms 时间复杂度o(1)
console.time('rotate2')

 

4、单元测试

// Jest
import {rotate1, rotate2} from '../把一个数组旋转看k步'

describe('数组旋转',()=>{
    it('正常情况', ()=> {
        const arr = [1,2,3,4,5,6,7]
        const k = 3
        const res = rotate1(arr, k)
        // const res = rotate2(arr, k)
        expect(res).toEqual([5,6,7,1,2,3,4]) // 断言
    })

    it('数组为空', ()=> {
        const res = rotate1([], 3)
        expect(res).toEqual([]) 
    })

    it('k是负值', ()=> {
        const arr = [1,2,3,4,5,6,7]
        const k = -3
        const res = rotate1(arr, k)
        expect(res).toEqual([5,6,7,1,2,3,4]) 
    })

    it('k是0', ()=> {
        const arr = [1,2,3,4,5,6,7]
        const k = 0
        const res = rotate1(arr, k)
        expect(res).toEqual(arr)
    })

    it('k不是数字', ()=> {
        const arr = [1,2,3,4,5,6,7]
        const k = 'abc'
        
        // @ts-ignore   // 报错注释
        const res = rotate1(arr, k)
        expect(res).toEqual(arr)
    })
})