【C++刷题】力扣-#566-重塑矩阵

题目描述

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

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

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

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

示例

示例 1

bash 复制代码
输入:nums = [[1,2],[3,4]], r = 1, c = 4
输出:[[1,2,3,4]]

示例 2

bash 复制代码
输入:nums = [[1,2],[3,4]], r = 2, c = 4
输出:[[1,2],[3,4]]
限制条件:
● 1 <= m, n, r, c <= 100
● m * n == r * c

题解

这个问题可以通过将原始矩阵展平为一维数组,然后再重新塑形为新的行和列数。

  1. 展平矩阵:遍历原始矩阵,将所有元素放入一个一维数组中。
  2. 重塑矩阵:根据新的行数 r 和列数 c,重新塑形一维数组为新的二维矩阵。

代码实现

cpp 复制代码
vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
    int m = mat.size(), n = mat[0].size(); // 获取原始矩阵的行数和列数
    int total = m * n;                     // 计算总元素数量
    if (total != r * c) {                  // 检查重塑操作是否合理
        return mat; // 如果不合理,返回原始矩阵
    }

    vector<int> flatNums(total); // 创建一个一维数组来存储所有元素
    int k = 0;
    for (const auto& row : mat) {
        for (int num : row) {
            flatNums[k++] = num;
        }
    }

    vector<vector<int>> reshaped(r, vector<int>(c)); // 创建新的矩阵
    for (int i = 0; i < total; ++i) {
        reshaped[i / c][i % c] = flatNums[i]; // 填充新的矩阵
    }
    return reshaped;
}

复杂度分析

● 时间复杂度:O(m * n),其中 m 和 n 分别是原始矩阵的行数和列数。我们需要遍历原始矩阵的所有元素。

● 空间复杂度:O(m * n),用于存储展平后的一维数组。

这个算法的优势在于它直接使用简单的遍历和数组操作来重塑矩阵,而不需要复杂的数学计算。

相关推荐
mrbone1118 分钟前
Git-git worktree的使用
开发语言·c++·git·cmake·worktree·gitab
凌肖战26 分钟前
力扣网C语言编程题:快慢指针来解决 “寻找重复数”
c语言·算法·leetcode
虾球xz2 小时前
CppCon 2018 学习:EFFECTIVE REPLACEMENT OF DYNAMIC POLYMORPHISM WITH std::variant
开发语言·c++·学习
津津有味道3 小时前
Qt C++串口SerialPort通讯发送指令读写NFC M1卡
linux·c++·qt·串口通信·serial·m1·nfc
让我们一起加油好吗3 小时前
【C++】list 简介与模拟实现(详解)
开发语言·c++·visualstudio·stl·list
傅里叶的耶3 小时前
C++系列(二):告别低效循环!选择、循环、跳转原理与优化实战全解析
c++·visual studio
Vitta_U4 小时前
MFC的List Control自适应主界面大小
c++·list·mfc
szekl4 小时前
HDMI 2.0 4×2矩阵切换器412HN——多信号输入输出的高清解决方案
linux·矩阵·计算机外设·电脑·ekl
Dovis(誓平步青云)5 小时前
基于探索C++特殊容器类型:容器适配器+底层实现原理
开发语言·c++·queue·适配器·stack
pipip.6 小时前
UDP————套接字socket
linux·网络·c++·网络协议·udp