Leetcode-初级算法系列-数组-旋转矩阵

题目描述

给定一个数组，将数组中的元素向右移动 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]

方法一 使用额外的数组

解题思路

对于原数组，每一个元素移动之后的位置为(i+k) mod n，在额外数组中将每个元素放到正确的位置，然后将额外数组拷贝到原数组中。

代码

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> array(n);
        for(int i = 0 ; i < n ; i++){
            array[(i+k) % n] = nums[i];
        }
        nums.assign(array.begin(),array.end());
    }
};

复杂度分析

时间复杂度：O(n)，n为数组的长度
空间复杂度：O(n)

方法二 环状替换

解题思路

方法一会使用到额外数组。如果我们直接在原数组中替换元素，会存在一个问题，即被替换的元素还没有被移动就被覆盖了，因此考虑交换两个元素，当元素回到最初的起点时，可能还没有遍历到原数组的所有元素，这时从下一个元素开始遍历，总共需要开始遍历的元素个数为gcd(n, k)。

代码

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        int n = nums.size();
        k = k % n;
        int count = gcd(k,n );
        for(int start = 0 ; start < count; ++start){
            int cur = start;
            int prev = nums[start]; //每次都把被覆盖的元素存储在当前循环的第一个元素中
            do{
                int next = (cur + k) % n;
                swap(nums[next] , prev);
                cur = next;
            }while(start != cur);
        }
    }
};

复杂度分析

时间复杂度：O(n),每个元素都会被遍历一次
空间复杂度: O(1),只需要常数的空间

方法三 翻转矩阵

解题思路

1.先将原矩阵全部翻转，得到A；
2.将矩阵A前K个元素进行翻转；
3.将矩阵A剩余元素进行翻转；

代码

class Solution {
public:
    void rotate(vector<int>& nums, int k) {
        k = k % nums.size();
        reverse(nums.begin(),nums.end());
        reverse(nums.begin(),nums.begin()+k);
        reverse(nums.begin()+k,nums.end());

    }
};