leetcode 566. Reshape the Matrix

最新推荐文章于 2021-05-22 20:29:00 发布

「已注销」

最新推荐文章于 2021-05-22 20:29:00 发布

阅读量109

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/wangchy29/article/details/89243469

版权

leetcode 专栏收录该内容

61 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        vector<vector<int>> res(r);
        int m=nums.size();
        int n = nums[0].size();
        if(m*n!=r*c)return nums;
        int row=0,col = 0,count = 0;
        for(int i=0;i<m;i++)
        {
            for(int j = 0;j<n;j++)
            {
                if(count==c)
                {
                    count = 0;
                    row++;
                }
                res[row].push_back(nums[i][j]);
                count++;
            }
        }
        return res;
    }
};