⭐算法入门⭐《坐标转换》简单01 —— LeetCode 566. 重塑矩阵

最新推荐文章于 2022-08-09 10:46:27 发布

英雄哪里出来

最新推荐文章于 2022-08-09 10:46:27 发布

阅读量1.4k

点赞数 12

分类专栏：《LeetCode算法全集》文章标签： c++ 算法数据结构坐标转换取模

本文链接：https://blog.csdn.net/WhereIsHeroFrom/article/details/118963877

版权

《LeetCode算法全集》专栏收录该内容

265 篇文章 1917 订阅

订阅专栏

🙉饭不食，水不饮，题必须刷🙉
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
LeetCode 太难？先看简单题！ 🧡《C语言入门100例》🧡
数据结构难？不存在的！ 🌳《数据结构入门》🌳
LeetCode 太简单？算法学起来！ 🌌《夜深人静写算法》🌌

一、题目

1、题目描述

给出一个由二维数组表示的矩阵，以及两个正整数 $r$ 和 $c$ ，分别表示想要的重构的矩阵的行数和列数。重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。如果具有给定参数重构的操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。
样例输入： $n u m s = [[1, 2], [3, 4]], r = 1, c = 4$
样例输出： $[[1, 2, 3, 4]]$

2、基础框架

c++ 版本给出的基础框架代码如下：

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
    }
};

输入数组参数类型是vector<vector<int>>&，代表的是一个引用类型的二维数组，按照题目要求，需要返回一个二维数组；

3、原题链接

LeetCode 566. 重塑矩阵

二、解题报告

1、思路分析

首先，原矩阵的个数统计出来，和给出的参数的乘积 $\times c$ 比较，不相等则输出原矩阵。
否则，我们来看个图：
可以将原先的二维矩阵通过行遍历转换成一维的，然后再根据 $r$ 和 $c$ ，每 $r$ 个作为新矩阵的行，进行重新填充。
对于一个 $\times m$ 的矩阵，我们可以从左往右，从上往下进行编号，令一维编号为 $i d$ ，
二维转一维：
$\times m + c$
一维转二维：
$\lfloor \frac {id} m \rfloor \\ c = {id} \ mod \ m$

2、时间复杂度

时间复杂度为 $O (n m)$ 。

3、代码详解

class Solution {
public:
    vector<vector<int>> matrixReshape(vector<vector<int>>& mat, int r, int c) {
        vector<vector<int>> ans;
        int n = mat.size();                        // (1)
        int m = mat[0].size();                     // (2)
        if(n*m != r*c) {
            return mat;
        }
        for(int i = 0; i < r; ++i) {
            vector <int> v;
            for(int j = 0; j < c; ++j) {
                int id = i * c + j;                // (3)
                v.push_back(mat[id / m][id % m]);  // (4)
            }
            ans.push_back(v);                      // (5)
        }
        return ans;
    }
};