美文网首页
leetcode算法练习- 旋转数组

leetcode算法练习- 旋转数组

作者: 土豪码农 | 来源:发表于2019-03-19 07:53 被阅读0次

    给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。

    示例 1:

    输入: [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:

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

    说明:
    尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
    要求使用空间复杂度为 O(1) 的原地算法。

    第一次提交

    想法非常简单,就是移动一次我就从数组尾部删除一个加到数组前

    var rotate = function(nums, k) {
        for(let i=0;i<k;i++){
            const num = nums.pop();
            nums.unshift(num)
        }
    };
    
    QQ图片20190319074118.png

    第二次提交

    优化了一下,数组长度为1或者数组长度和k相同的直接返回,移动步数大于长度的可以求余节省步数,最后不用一个个操作,直接把移动多少用splice切出来再塞前面

      var rotate = function (nums, k) {
        const length = nums.length;
        if (length === k || length === 1) {
          return nums
        } else {
          if (length < k) {
            k = k % length
          }
          nums.unshift(...nums.splice( - k, k));
        }
      };
    
    QQ图片20190319075412.png

    最优解

    看了下最优解,和我第二次提交用的方法基本是一样的

    var rotate = function(nums, k) {
        let l = nums.length;
        k %= l
        nums.unshift(...nums.splice(nums.length - k, k))
    };
    

    相关文章

      网友评论

          本文标题:leetcode算法练习- 旋转数组

          本文链接:https://www.haomeiwen.com/subject/iljwmqtx.html