LeetCode 每日一题 566重塑矩阵

最新推荐文章于 2025-12-06 19:00:36 发布

原创最新推荐文章于 2025-12-06 19:00:36 发布 · 74 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #leetcode

该博客内容涉及C++编程，主要讲解了一个名为`Solution`的类中`matrixReshape`函数的实现，该函数用于将给定的矩阵进行重塑。函数接受一个二维整数数组（矩阵）和新的行数及列数参数，当矩阵元素总数能被新的行数和列数整除时，函数返回重塑后的矩阵；否则，返回原始矩阵。

分析

按照题意模拟过程

C++ 代码

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> temp;
    vector<vector<int>> matrixReshape(vector<vector<int>>& nums, int r, int c) {
        int n=nums.size(),m=nums[0].size();
        if(n*m!=r*c) return nums;
        for(auto row:nums)
        {
            for(auto x:row)
            {
                temp.push_back(x);
                if(temp.size()==c)
                {
                    ans.push_back(temp);
                    temp.clear();
                }
            }
        }
        return ans;
    }
};