leetcode566_2-17每日题：重塑矩阵

恰似惊鸿踏雪泥

于 2021-02-17 11:35:25 发布

阅读量210

点赞数

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/SJTU_liangge/article/details/113832486

版权

leetcode 专栏收录该内容

105 篇文章

订阅专栏

该代码实现了一个矩阵重塑的功能，将给定的二维矩阵按照指定的行数和列数进行重塑。首先检查原始矩阵的元素数量是否能匹配新的行数和列数，如果可以则进行重塑，否则返回原始矩阵。代码使用了双层循环遍历并重构矩阵。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        int x=nums.size();
        int y;
        if(x) y=nums[0].size();
        else return nums;
        if(x*y!=r*c) return nums;
        vector<vector<int> > answer;
        vector<int> now;
        int i,j;
        for(i=0;i<x;i++){
            for(j=0;j<y;j++){
                now.push_back(nums[i][j]);
                if(now.size()==c){
                    answer.push_back(now);
                    now.clear();
                } 
            }
        }
        return answer;
    }
};

在这里插入图片描述