[LeetCode]566. Reshape the Matrix

来源:互联网 发布:电脑软件开发学校 编辑:程序博客网 时间:2024/06/05 17:04

Solution如下

class Solution {public:    vector<vector<int>> matrixReshape(vector<vector<int>> &nums, int r, int c)    {        if (nums.size() == 0 && r * c != 0) {            return nums;        }        if (nums.size() * nums[0].size() != r * c) {            return nums;        }        vector<vector<int>> ret(r, vector<int>());        cout << ret.size() << " " << ret[0].size() << endl;        int index = 0;        for (auto sub : nums) {            for (auto i : sub) {                                ret[index / c].push_back(i);                index++;                if (index == r * c)                    return ret;            }        }        return ret;    }};
原创粉丝点击