566. 重塑矩阵

最新推荐文章于 2023-09-10 14:44:19 发布

无敌的神龙战士

最新推荐文章于 2023-09-10 14:44:19 发布

阅读量448

点赞数

CC 4.0 BY-SA版权

分类专栏： # STL

Tzt的打工时间

本文链接：https://blog.csdn.net/weixin_43599304/article/details/119748932

STL 专栏收录该内容

29 篇文章

订阅专栏

这篇博客介绍了MATLAB中的reshape函数，该函数用于将mxn矩阵重塑为rxc的新矩阵，同时保持原始数据不变。文章通过示例解释了如何使用reshape，并展示了当重塑条件允许时如何重构矩阵，以及在条件不满足时返回原始矩阵的情况。

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

重塑矩阵
在 MATLAB 中，有一个非常有用的函数 reshape ，它可以将一个 m x n 矩阵重塑为另一个大小不同（r x c）的新矩阵，但保留其原始数据。

给你一个由二维数组 mat 表示的 m x n 矩阵，以及两个正整数 r 和 c ，分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

如果具有给定参数的 reshape 操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

示例 1：

输入：mat = [[1,2],[3,4]], r = 1, c = 4
输出：[[1,2,3,4]]

示例 2：


输入：mat = [[1,2],[3,4]], r = 2, c = 4
输出：[[1,2],[3,4]]

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
        
        int m = mat.size(), n = mat[0].size();
        if(m * n == r * c){
            vector<vector<int>> result(r, vector<int>(c));
            for(int i = 0; i < m * n; ++i){
                result[i / c][i % c] = mat[i / n][i % n];   
            }
            return result;
        }
        return mat;
    }
};