Leetcode：1260. 二维网格迁移

class Solution {
public:
    vector<int>shiftRight(vector<int>v, int k)
    {
        reverse(v.begin(),v.end());
        reverse(v.begin(),v.begin()+k);
        reverse(v.begin()+1+k,v.end());
        return v;
    }
    vector<vector<int>> shiftGrid(vector<vector<int>>& grid, int k) {

        

        //先将grid中的元素转移至向量中
        vector<int>temp(grid.size()*grid[0].size(),0);
        for(int i=0;i<grid.size();i++)
        {
            for(int j=0;j<grid[0].size();j++)
            {
                temp[(i*grid[0].size()+j+k)%(grid.size()*grid[0].size())]=grid[i][j];
            }
        }
        k=k%temp.size();///注意这里一定要具有取余操作，不然会导致溢出

        //对一位矩阵进行迁移
        shiftRight(temp,k);
        //
        for(int i=0;i<grid.size();i++)
        {
            for(int j=0;j<grid[0].size();j++)
            {
                grid[i][j]=temp[i*grid[0].size()+j];
            }
        }

        return grid;
    }
};