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;
}
};
Leetcode:1260. 二维网格迁移
最新推荐文章于 2022-07-20 20:39:42 发布