leetcode每日一练、数据结构（566、重塑矩阵）-优快云博客

本文链接：https://blog.youkuaiyun.com/m0_69808040/article/details/128878796

该文章介绍了一个在Java中实现MATLAB的reshape函数的示例，用于将mxn矩阵重塑为rxc的新矩阵，保持原始数据。代码通过行和列的映射关系进行元素重排，如果重塑条件合理则返回新矩阵，否则返回原矩阵。

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

题目：

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

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

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

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

思路：

考虑二维数组中坐标的一维表示，二维数组中的行坐标对应“/”操作，二维数组中的列坐标对应“%”操作，通过这两种映射方式将给出数组中的元素移动到目标数组中即可

java代码

package demo;

public class leetcode7 {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int rowcount=mat.length;
        int colcount=0;
        int[][] result=new int[r][c];
        if(rowcount>0){
            colcount=mat[0].length;
        }
        if(r*c!=rowcount*colcount)
            return mat;
        for(int i=0;i<r*c;i++){
            result[i/c][i%c]=mat[i/colcount][i%colcount];
        }
        return result;
    }
}

class test7{
    public static void main(String arg[]){
        leetcode7 l=new leetcode7();
        int[][] nums={{1,2},{3,4}};
        int r=1;
        int c=4;
        int[][] result=l.matrixReshape(nums,r,c);
        for(int i=0;i<r;i++){
            for(int j=0;j<c;j++){
                System.out.println(result[i][j]);
            }
        }

    }
}