将二维数组转换成一维数组,再实现一维转二维

设 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 shiftGrid(vector>& grid) {
        int m = grid.size(); //行数
        int n = grid[0].size(); //列数
        vector 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> shiftGrid(vector& tmp) {
        int m = grid.size(); //行数
        int n = grid[0].size(); //列数
        vector grid(m, vector(n)); 
        for(int i = 0 ; i < tmp.size(); i++){
             grid[i / n][i % n] = tmp[i];
        }
        return grid;
    }
};

你可能感兴趣的:(算法,python,数据结构,numpy)