重塑矩阵

/*
 * Problem: Reshape a given m x n matrix to a new r x c matrix while maintaining the original row-wise order.
 * This solution uses Java Streams to achieve the same.
 */
import java.util.*;
import java.util.stream.*;
 
public class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int m = mat.length, n = mat[0].length;
        // Check if reshaping is possible
        if (m * n != r * c) {
            return mat;
        }
        // Flatten the matrix and collect elements
        int[] flatArray = Arrays.stream(mat)
                                .flatMapToInt(Arrays::stream)
                                .toArray();
        // Create the new reshaped matrix
        int[][] reshapedMatrix = new int[r][c];
        for (int i = 0; i < r * c; i++) {
            reshapedMatrix[i / c][i % c] = flatArray[i];
        }
        return reshapedMatrix;
    }
}