【LeetCode】566 Reshape the Matrix

原题地址：566. Reshape the Matrix。

解法一

vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {    int len1 = nums.size();    int len2 = nums[0].size();    vector<vector<int>> res;    vector<int> line;    if (len1 * len2 == r * c) {        for (int i = 0; i < len1; i++) {            for (int j = 0; j < len2; j++) {                line.push_back(nums[i][j]);            }        }        for (int i = 0; i < r; i++) {            vector<int> temp;            for (int j = 0; j < c; j++) {                temp.push_back(line[i * c + j]);            }            res.push_back(temp);        }    }    else {        res = nums;    }    return res;}

解法二

vector<vector<int>> matrixReshape2(vector<vector<int>>& nums, int r, int c) {    int len1 = nums.size();    int len2 = nums[0].size();    vector<vector<int>> res(r, vector<int>(c, 0));    if (len1 * len2 == r * c) {        for (int i = 0; i < r; i++) {            for (int j = 0; j < c; j++) {                int k = i * c + j;                res[i][j] = nums[k / len2][k % len2];            }        }    }    else {        res = nums;    }    return res;}

参考：

[C++] Clean Code - 5 lines (2 Solution)

最后更新于2017年6月19日。