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;
    }
};