[leetCode刷题笔记]566. Reshape the Matrix

原创于 2017-06-06 11:24:40 发布 · 304 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一个用于将二维矩阵重塑为另一个指定大小的新矩阵的算法。该算法通过遍历原始矩阵的元素，并按照行优先的方式将其填充到新矩阵中，从而实现矩阵形状的变化。此方法适用于当目标矩阵的元素数量等于原始矩阵的元素数量时。

public class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        if (nums == null || nums.length == 0 || r * c != nums.length * nums[0].length) return nums;
        int[][] result = new int[r][c];
        
        int k = 0;
        
        
        for (int i = 0; i < nums.length; i++ ) {
            for (int j = 0; j < nums.length; j++) {
                result[k / c][k % c] = nums[i][j];
                k++;
            }
        }
        return result;
        
    }
}