设 m 和 n 分别为网格的行列数,grid[m][n]是一个二维数组,那么如何将其转化成一维数组?
假设我们要得到grid[i][j]在一维数组中的位置:index = i * n + j (意思就是一共有i行,每行n个元素,然后再加上第j列,也就是有j个元素)。这样就可以得到一维数组的下标位置。
那么如何逆向转换?
这个就更加简单了。首先考虑将index 拆分成每一行有n个元素,就转换成均分的问题 index / n 就得到在第几行,在第几列就可以求得除完之后的余数,我们使用 index % n 就能得到余数了。
代码实现
二维转一维:
//二维转一维
class Solution {
public:
vector<int> shiftGrid(vector<vector<int>>& grid) {
int m = grid.size(); //行数
int n = grid[0].size(); //列数
vector<int> tmp(n * m);
for(int i = 0 ; i < m; i++){
for(int j = 0; j < n; j++){
//转换成一维数组的下标
int index = i * n + j;
tmp[index] = grid[i][j];
}
}
return tmp;
}
};
一维转二维:
//一维转二维
class Solution {
public:
vector<vector<int>> shiftGrid(vector<int>& tmp) {
int m = grid.size(); //行数
int n = grid[0].size(); //列数
vector<int> grid(m, vector<int>(n));
for(int i = 0 ; i < tmp.size(); i++){
grid[i / n][i % n] = tmp[i];
}
return grid;
}
};