关于操作矩阵中索引的技巧(二)

来源:互联网 发布:mac分屏功能怎么用 编辑:程序博客网 时间:2024/06/06 09:31

关于矩阵的操作,做了一些题发现,对于一个m*n的矩阵,这个类行多少列n是十分关键的,因为我拿到一个累计长度k,则通过除法和取余操作,就可以得到这个数在矩阵中的索引位置:

int i = k / n;

int j = k % n;


比如要对一个矩阵操作,reshape等,可以采用如下的算法来完成:

vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
int m = nums.size(), n = nums[0].size();
if (m * n != r * c) return nums;
vector<vector<int>> M(r, vector<int>(c, 0));
int k = 0;
while (k < m * n) {
M[k/c][k%c] = nums[k/n][k%n];
k++;
}
return M;
}

有一个计数k,从0一直到m*n,每次只增加1,这样通过对k做索引映射,便可以得到在矩阵中的位置信息,接着操作便可以了,这是一个矩阵操作的模板,可以记下来,以后就这么用就可以了。

0 0